Hugging Face's logo

Spaces:
Vedant104
/
Spc2
Sleeping

App Files Community
Vedant104 commited on
Commit
5a4adc9
·
verified ·
1 Parent(s): b1b55f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,7 +35,8 @@ pipe = pipeline(
35
  "text-generation",
36
  # model="Qwen/Qwen2.5-0.5B-Instruct",
37
  model="Qwen/Qwen2.5-1.5B-Instruct",
38
- device="cpu"
 
39
  )
40
 
41
  # =========================
 
35
  "text-generation",
36
  # model="Qwen/Qwen2.5-0.5B-Instruct",
37
  model="Qwen/Qwen2.5-1.5B-Instruct",
38
+ device="cpu",
39
+ torch_dtype="auto"
40
  )
41
 
42
  # =========================