Hugging Face's logo

Spaces:
GIGAParviz
/
Parviz_Mind
Sleeping

App Files
GIGAParviz commited on
Commit
cf98681
·
verified ·
1 Parent(s): 03fdc98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ pipeline = transformers.pipeline(
10
  "text-generation",
11
  model=model_id,
12
  model_kwargs={"torch_dtype": torch.bfloat16},
13
- device="auto",
14
  )
15
 
16
 
 
10
  "text-generation",
11
  model=model_id,
12
  model_kwargs={"torch_dtype": torch.bfloat16},
13
+ device="cpu",
14
  )
15
 
16