Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ CSS = """
|
|
47 |
|
48 |
|
49 |
tokenizer = AutoTokenizer.from_pretrained("shenzhi-wang/Llama3-8B-Chinese-Chat")
|
50 |
-
model = AutoModelForCausalLM.from_pretrained("shenzhi-wang/Llama3-8B-Chinese-Chat", device_map="auto")
|
51 |
|
52 |
|
53 |
@spaces.GPU
|
|
|
47 |
|
48 |
|
49 |
tokenizer = AutoTokenizer.from_pretrained("shenzhi-wang/Llama3-8B-Chinese-Chat")
|
50 |
+
model = AutoModelForCausalLM.from_pretrained("shenzhi-wang/Llama3-8B-Chinese-Chat", torch_dtype="auto", device_map="auto")
|
51 |
|
52 |
|
53 |
@spaces.GPU
|