summaryrefslogtreecommitdiffstats
path: root/g4f
diff options
context:
space:
mode:
authorkqlio67 <kqlio67@users.noreply.github.com>2024-09-11 14:54:26 +0200
committerkqlio67 <kqlio67@users.noreply.github.com>2024-09-11 14:54:26 +0200
commit7d1fa2c623db61bb9116abc8a059d464324b41c9 (patch)
tree4bc28439337e1f07e1b511d2d8757e58c4f08b95 /g4f
parentrefactor(g4f/Provider/AiChatOnline.py): streamline class attributes (diff)
downloadgpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.tar
gpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.tar.gz
gpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.tar.bz2
gpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.tar.lz
gpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.tar.xz
gpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.tar.zst
gpt4free-7d1fa2c623db61bb9116abc8a059d464324b41c9.zip
Diffstat (limited to 'g4f')
-rw-r--r--g4f/Provider/ReplicateHome.py16
1 files changed, 4 insertions, 12 deletions
diff --git a/g4f/Provider/ReplicateHome.py b/g4f/Provider/ReplicateHome.py
index c4e52ad6..d41633ba 100644
--- a/g4f/Provider/ReplicateHome.py
+++ b/g4f/Provider/ReplicateHome.py
@@ -15,16 +15,11 @@ class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
parent = "Replicate"
working = True
default_model = 'meta/meta-llama-3-70b-instruct'
+ text_models = {"meta/meta-llama-3-70b-instruct", "mistralai/mixtral-8x7b-instruct-v0.1", "google-deepmind/gemma-2b-it"}
+ image_models = {"stability-ai/stable-diffusion-3", "bytedance/sdxl-lightning-4step", "playgroundai/playground-v2.5-1024px-aesthetic"}
models = [
- # Models for image generation
- 'stability-ai/stable-diffusion-3',
- 'bytedance/sdxl-lightning-4step',
- 'playgroundai/playground-v2.5-1024px-aesthetic',
-
- # Models for image generation
- 'meta/meta-llama-3-70b-instruct',
- 'mistralai/mixtral-8x7b-instruct-v0.1',
- 'google-deepmind/gemma-2b-it',
+ *text_models,
+ *image_models
]
versions = {
@@ -51,9 +46,6 @@ class ReplicateHome(AsyncGeneratorProvider, ProviderModelMixin):
]
}
- image_models = {"stability-ai/stable-diffusion-3", "bytedance/sdxl-lightning-4step", "playgroundai/playground-v2.5-1024px-aesthetic"}
- text_models = {"meta/meta-llama-3-70b-instruct", "mistralai/mixtral-8x7b-instruct-v0.1", "google-deepmind/gemma-2b-it"}
-
model_aliases = {
"sd-3": "stability-ai/stable-diffusion-3",
"sdxl": "bytedance/sdxl-lightning-4step",