Spaces:
Running
on
Zero
Running
on
Zero
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"
|