diff options
| author | kqlio67 <kqlio67@users.noreply.github.com> | 2024-09-11 14:54:26 +0200 |
|---|---|---|
| committer | kqlio67 <kqlio67@users.noreply.github.com> | 2024-09-11 14:54:26 +0200 |
| commit | 7d1fa2c623db61bb9116abc8a059d464324b41c9 (patch) | |
| tree | 4bc28439337e1f07e1b511d2d8757e58c4f08b95 /g4f/Provider/ReplicateHome.py | |
| parent | refactor(g4f/Provider/AiChatOnline.py): streamline class attributes (diff) | |
| download | gpt4free-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 '')
| -rw-r--r-- | g4f/Provider/ReplicateHome.py | 16 |
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", |