Spaces:

han1997
/

cobra

Runtime error

han1997 commited on Mar 24

Commit

de7186d

•

1 Parent(s): 9cc868e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ vlm = load("cobra+3b")
 if torch.cuda.is_available():
     DEVICE = "cuda"
-    DTYPE = torch.float16
 else:
     DEVICE = "cpu"
     DTYPE = torch.float32

 if torch.cuda.is_available():
     DEVICE = "cuda"
+    DTYPE = torch.float32
 else:
     DEVICE = "cpu"
     DTYPE = torch.float32