Hugging Face's logo

Spaces:
OFA-Sys
/
OFA-Generic_Interface
Build error

App Files Community
6
JustinLin610 commited on
Commit
6edfbf7
•
1 Parent(s): 31438ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ tasks.register_task('vqa_gen', VqaGenTask)
51
  # turn on cuda if GPU is available
52
  use_cuda = torch.cuda.is_available()
53
  # use fp16 only when GPU is available
54
- use_fp16 = False
55
 
56
  # download checkpoints
57
  os.system('wget https://ofa-silicon.oss-us-west-1.aliyuncs.com/checkpoints/caption_demo.pt; '
 
51
  # turn on cuda if GPU is available
52
  use_cuda = torch.cuda.is_available()
53
  # use fp16 only when GPU is available
54
+ use_fp16 = True
55
 
56
  # download checkpoints
57
  os.system('wget https://ofa-silicon.oss-us-west-1.aliyuncs.com/checkpoints/caption_demo.pt; '