XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Update (g4f/models.py g4f/Provider/ReplicateHome.py)

330d920e
kqlio67 <kqlio67@users.noreply.github.com>
提交于

代码差异

2 个文件 +19 -31
Modified g4f/Provider/ReplicateHome.py +17 -15
@@ -17,7 +17,13 @@ class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
17 17 supports_system_message = True
18 18 supports_message_history = True
19 19
20 default_model = 'meta/meta-llama-3-70b-instruct'
20 default_model = 'yorickvp/llava-13b'
21
22 image_models = [
23 'stability-ai/stable-diffusion-3',
24 'bytedance/sdxl-lightning-4step',
25 'playgroundai/playground-v2.5-1024px-aesthetic',
26 ]
21 27
22 28 text_models = [
23 29 'meta/meta-llama-3-70b-instruct',
@@ -26,35 +32,31 @@ class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
26 32 'yorickvp/llava-13b',
27 33 ]
28 34
29 image_models = [
30 'black-forest-labs/flux-schnell',
31 'stability-ai/stable-diffusion-3',
32 'bytedance/sdxl-lightning-4step',
33 'playgroundai/playground-v2.5-1024px-aesthetic',
34 ]
35
35 36
36 37 models = text_models + image_models
37 38
38 39 model_aliases = {
39 "flux-schnell": "black-forest-labs/flux-schnell",
40 # image_models
40 41 "sd-3": "stability-ai/stable-diffusion-3",
41 42 "sdxl": "bytedance/sdxl-lightning-4step",
42 43 "playground-v2.5": "playgroundai/playground-v2.5-1024px-aesthetic",
43 "llama-3-70b": "meta/meta-llama-3-70b-instruct",
44 "mixtral-8x7b": "mistralai/mixtral-8x7b-instruct-v0.1",
44
45 # text_models
45 46 "gemma-2b": "google-deepmind/gemma-2b-it",
46 47 "llava-13b": "yorickvp/llava-13b",
47 48 }
48 49
49 50 model_versions = {
50 "meta/meta-llama-3-70b-instruct": "fbfb20b472b2f3bdd101412a9f70a0ed4fc0ced78a77ff00970ee7a2383c575d",
51 "mistralai/mixtral-8x7b-instruct-v0.1": "5d78bcd7a992c4b793465bcdcf551dc2ab9668d12bb7aa714557a21c1e77041c",
52 "google-deepmind/gemma-2b-it": "dff94eaf770e1fc211e425a50b51baa8e4cac6c39ef074681f9e39d778773626",
53 "yorickvp/llava-13b": "80537f9eead1a5bfa72d5ac6ea6414379be41d4d4f6679fd776e9535d1eb58bb",
54 'black-forest-labs/flux-schnell': "f2ab8a5bfe79f02f0789a146cf5e73d2a4ff2684a98c2b303d1e1ff3814271db",
51 # image_models
55 52 'stability-ai/stable-diffusion-3': "527d2a6296facb8e47ba1eaf17f142c240c19a30894f437feee9b91cc29d8e4f",
56 53 'bytedance/sdxl-lightning-4step': "5f24084160c9089501c1b3545d9be3c27883ae2239b6f412990e82d4a6210f8f",
57 54 'playgroundai/playground-v2.5-1024px-aesthetic': "a45f82a1382bed5c7aeb861dac7c7d191b0fdf74d8d57c4a0e6ed7d4d0bf7d24",
55
56 # text_models
57 "google-deepmind/gemma-2b-it": "dff94eaf770e1fc211e425a50b51baa8e4cac6c39ef074681f9e39d778773626",
58 "yorickvp/llava-13b": "80537f9eead1a5bfa72d5ac6ea6414379be41d4d4f6679fd776e9535d1eb58bb",
59
58 60 }
59 61
60 62 @classmethod
Modified g4f/models.py +2 -16
@@ -164,12 +164,6 @@ llama_3_8b = Model(
164 164 best_provider = IterListProvider([Cloudflare])
165 165 )
166 166
167 llama_3_70b = Model(
168 name = "llama-3-70b",
169 base_provider = "Meta Llama",
170 best_provider = IterListProvider([ReplicateHome])
171 )
172
173 167 # llama 3.1
174 168 llama_3_1_8b = Model(
175 169 name = "llama-3.1-8b",
@@ -212,7 +206,7 @@ mistral_7b = Model(
212 206 mixtral_8x7b = Model(
213 207 name = "mixtral-8x7b",
214 208 base_provider = "Mistral",
215 best_provider = IterListProvider([DDG, ReplicateHome])
209 best_provider = DDG
216 210 )
217 211
218 212 mistral_nemo = Model(
@@ -279,7 +273,7 @@ gemini = Model(
279 273 gemma_2b = Model(
280 274 name = 'gemma-2b',
281 275 base_provider = 'Google',
282 best_provider = IterListProvider([ReplicateHome])
276 best_provider = ReplicateHome
283 277 )
284 278
285 279
@@ -583,12 +577,6 @@ flux_4o = Model(
583 577
584 578 )
585 579
586 flux_schnell = Model(
587 name = 'flux-schnell',
588 base_provider = 'Flux AI',
589 best_provider = ReplicateHome
590
591 )
592 580
593 581
594 582 ### Other ###
@@ -635,7 +623,6 @@ class ModelUtils:
635 623
636 624 # llama-3
637 625 'llama-3-8b': llama_3_8b,
638 'llama-3-70b': llama_3_70b,
639 626
640 627 # llama-3.1
641 628 'llama-3.1-8b': llama_3_1_8b,
@@ -802,7 +789,6 @@ class ModelUtils:
802 789 'flux-disney': flux_disney,
803 790 'flux-pixel': flux_pixel,
804 791 'flux-4o': flux_4o,
805 'flux-schnell': flux_schnell,
806 792
807 793
808 794 ### Other ###