Spaces:
Build error
Build error
Update app.py
Browse files
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 =
|
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 = False
|
55 |
|
56 |
# download checkpoints
|
57 |
os.system('wget https://ofa-silicon.oss-us-west-1.aliyuncs.com/checkpoints/caption_demo.pt; '
|