返回提交历史
Modified
g4f/Provider/Cloudflare.py
+6
-6
XFEstudio/gpt4free
Update (g4f/Provider/Cloudflare.py)
c74a6943
代码差异
1 个文件
+6
-6
@@ -21,10 +21,7 @@ class Cloudflare(AsyncGeneratorProvider, ProviderModelMixin):
21
21
supports_message_history = True
22
22
23
23
default_model = '@cf/meta/llama-3.1-8b-instruct-awq'
24
models = [
25
'@cf/tiiuae/falcon-7b-instruct', # Specific answer
26
27
24
models = [
28
25
'@hf/google/gemma-7b-it',
29
26
30
27
'@cf/meta/llama-2-7b-chat-fp16',
@@ -120,9 +117,12 @@ class Cloudflare(AsyncGeneratorProvider, ProviderModelMixin):
120
117
121
118
scraper = cloudscraper.create_scraper()
122
119
120
121
prompt = messages[-1]['content']
122
123
123
data = {
124
124
"messages": [
125
{"role": "user", "content": format_prompt(messages)}
125
{"role": "user", "content": prompt}
126
126
],
127
127
"lora": None,
128
128
"model": model,
@@ -147,7 +147,7 @@ class Cloudflare(AsyncGeneratorProvider, ProviderModelMixin):
147
147
148
148
response.raise_for_status()
149
149
150
skip_tokens = ["</s>", "<s>", "[DONE]", "<|endoftext|>", "<|end|>"]
150
skip_tokens = ["</s>", "<s>", "</s>", "[DONE]", "<|endoftext|>", "<|end|>"]
151
151
filtered_response = ""
152
152
153
153
for line in response.iter_lines():