Hugging Face's logo

Spaces:
asv7j
/
ModelTest2
Runtime error

App Files Community
asv7j commited on
Commit
c8120a6
·
verified ·
1 Parent(s): af7530b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ model1 = AutoModelForCausalLM.from_pretrained(
16
  )
17
  model = AutoModelForCausalLM.from_pretrained(
18
  "Qwen/Qwen2-1.5B-Instruct",
19
- device_map="auto"
20
  torch_dtype="auto"
21
  )
22
 
 
16
  )
17
  model = AutoModelForCausalLM.from_pretrained(
18
  "Qwen/Qwen2-1.5B-Instruct",
19
+ device_map="auto",
20
  torch_dtype="auto"
21
  )
22