Spaces:

Akjava
/

chat-phi-4-deepseek-R1K-RL-EZO

Runtime error

Akjava commited on Sep 21, 2024

Commit

d66891a

1 Parent(s): d6cc9e4

update model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import gradio as gr
 text_generator = None
 is_hugging_face = True
 model_id = "google/gemma-2-9b-it"# too big
-model_id = "google/gemma-2-2b-it"
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
 device = "cuda"

 text_generator = None
 is_hugging_face = True
 model_id = "google/gemma-2-9b-it"# too big
+#model_id = "google/gemma-2-2b-it"
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
 device = "cuda"