Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,7 @@ import os
|
|
4 |
model = AutoModelForCausalLM.from_pretrained(
|
5 |
"Qwen/CodeQwen1.5-7B-Chat",
|
6 |
torch_dtype="auto",
|
7 |
-
device_map="auto"
|
8 |
-
token=hf_token
|
9 |
)
|
10 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/CodeQwen1.5-7B-Chat")
|
11 |
|
|
|
4 |
model = AutoModelForCausalLM.from_pretrained(
|
5 |
"Qwen/CodeQwen1.5-7B-Chat",
|
6 |
torch_dtype="auto",
|
7 |
+
device_map="auto"
|
|
|
8 |
)
|
9 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/CodeQwen1.5-7B-Chat")
|
10 |
|