Hugging Face's logo

Spaces:
Ketengan-Diffusion-Lab
/
Dolphin-Inference
Build error

App Files Community
Ketengan-Diffusion-Lab commited on
Commit
96bcfdf
·
verified ·
1 Parent(s): 429ad32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ transformers.logging.disable_progress_bar()
11
  warnings.filterwarnings('ignore')
12
 
13
  # set device
14
- torch.set_default_device('cuda') # or 'cpu'
15
 
16
  model_name = 'cognitivecomputations/dolphin-vision-7b'
17
 
 
11
  warnings.filterwarnings('ignore')
12
 
13
  # set device
14
+ torch.set_default_device('cuda:0') # or 'cpu'
15
 
16
  model_name = 'cognitivecomputations/dolphin-vision-7b'
17