Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ def init():
|
|
16 |
|
17 |
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
18 |
model_id = "google/gemma-2b"
|
|
|
19 |
|
20 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
21 |
#device = "cuda"
|
|
|
16 |
|
17 |
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
18 |
model_id = "google/gemma-2b"
|
19 |
+
model_id = "Qwen/Qwen2.5-0.5B-Instruct"
|
20 |
|
21 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
22 |
#device = "cuda"
|