返回提交历史
Modified
g4f/Provider/ReplicateHome.py
+4
-12
XFEstudio/gpt4free
refactor(g4f/Provider/ReplicateHome.py): optimize model organization and improve readability
7d1fa2c6
代码差异
1 个文件
+4
-12
@@ -15,16 +15,11 @@ class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
15
15
parent = "Replicate"
16
16
working = True
17
17
default_model = 'meta/meta-llama-3-70b-instruct'
18
text_models = {"meta/meta-llama-3-70b-instruct", "mistralai/mixtral-8x7b-instruct-v0.1", "google-deepmind/gemma-2b-it"}
19
image_models = {"stability-ai/stable-diffusion-3", "bytedance/sdxl-lightning-4step", "playgroundai/playground-v2.5-1024px-aesthetic"}
18
20
models = [
19
# Models for image generation
20
'stability-ai/stable-diffusion-3',
21
'bytedance/sdxl-lightning-4step',
22
'playgroundai/playground-v2.5-1024px-aesthetic',
23
24
# Models for image generation
25
'meta/meta-llama-3-70b-instruct',
26
'mistralai/mixtral-8x7b-instruct-v0.1',
27
'google-deepmind/gemma-2b-it',
21
*text_models,
22
*image_models
28
23
]
29
24
30
25
versions = {
@@ -51,9 +46,6 @@ class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
51
46
]
52
47
}
53
48
54
image_models = {"stability-ai/stable-diffusion-3", "bytedance/sdxl-lightning-4step", "playgroundai/playground-v2.5-1024px-aesthetic"}
55
text_models = {"meta/meta-llama-3-70b-instruct", "mistralai/mixtral-8x7b-instruct-v0.1", "google-deepmind/gemma-2b-it"}
56
57
49
model_aliases = {
58
50
"sd-3": "stability-ai/stable-diffusion-3",
59
51
"sdxl": "bytedance/sdxl-lightning-4step",