Spaces:

gokaygokay
/

FLUX-Prompt-Generator

Running on Zero

gokaygokay commited on Sep 23, 2024

Commit

00b5de2

1 Parent(s): edd2f71

groq models

Files changed (1) hide show

ui_components.py CHANGED Viewed

@@ -201,7 +201,7 @@ def create_interface():
         def update_model_choices(provider):
             provider_models = {
                 "Hugging Face": ["meta-llama/Meta-Llama-3.1-70B-Instruct"],
-                "Groq": ["llama-3.1-70b-versatile"],
                 "OpenAI": ["gpt-4o", "gpt-4o-mini"],
                 "Anthropic": ["claude-3-5-sonnet-20240620"],
                 "SambaNova": ["Meta-Llama-3.1-70B-Instruct"],

         def update_model_choices(provider):
             provider_models = {
                 "Hugging Face": ["meta-llama/Meta-Llama-3.1-70B-Instruct"],
+                "Groq": ["llama-3.1-70b-versatile", "mixtral-8x7b-32768", "gemma2-9b-it"],
                 "OpenAI": ["gpt-4o", "gpt-4o-mini"],
                 "Anthropic": ["claude-3-5-sonnet-20240620"],
                 "SambaNova": ["Meta-Llama-3.1-70B-Instruct"],