Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,7 @@ def init():
|
|
14 |
print("no HUGGINGFACE_TOKEN if you need set secret ")
|
15 |
#raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
|
16 |
|
17 |
-
model_id = "
|
18 |
-
model_id = "Qwen/Qwen2-0.5B-Instruct"
|
19 |
|
20 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
21 |
#device = "cuda"
|
|
|
14 |
print("no HUGGINGFACE_TOKEN if you need set secret ")
|
15 |
#raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
|
16 |
|
17 |
+
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
|
|
18 |
|
19 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
20 |
#device = "cuda"
|