Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def init():
|
|
16 |
#raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
|
17 |
|
18 |
model_id = "google/gemma-2-9b-it"
|
19 |
-
model_id = "microsoft/Phi-3-mini-128k-instruct"
|
20 |
|
21 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
22 |
device = "cuda"
|
|
|
16 |
#raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
|
17 |
|
18 |
model_id = "google/gemma-2-9b-it"
|
19 |
+
#model_id = "microsoft/Phi-3-mini-128k-instruct"
|
20 |
|
21 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
22 |
device = "cuda"
|