Hugging Face's logo

Spaces:
seawolf2357
/
cog5bvid
Runtime error

App Files Community
seawolf2357 commited on
Commit
fc71c2b
·
verified ·
1 Parent(s): c7a9993

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ from transformers import pipeline
29
  import gc
30
 
31
  # Set CUDA device and enable cuDNN
32
- torch.cuda.set_device(2)
33
  torch.backends.cudnn.enabled = True
34
 
35
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
29
  import gc
30
 
31
  # Set CUDA device and enable cuDNN
32
+ torch.cuda.set_device(0)
33
  torch.backends.cudnn.enabled = True
34
 
35
  device = "cuda" if torch.cuda.is_available() else "cpu"