APGASU commited on
Commit
eea565e
·
verified ·
1 Parent(s): 731043e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ CLASS_DROPOUT_PROB = 0.1
55
  CAPPING = 50.0
56
 
57
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
58
- DTYPE = torch.bfloat16 if DEVICE == "cuda" else torch.float32
59
  COMPILE = 0 #os.environ.get("VIBETOKEN_NO_COMPILE", "0") != "1" and DEVICE == "cuda"
60
 
61
  # ---------------------------------------------------------------------------
 
55
  CAPPING = 50.0
56
 
57
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
58
+ DTYPE = torch.float32 #torch.bfloat16 if DEVICE == "cuda" else torch.float32
59
  COMPILE = 0 #os.environ.get("VIBETOKEN_NO_COMPILE", "0") != "1" and DEVICE == "cuda"
60
 
61
  # ---------------------------------------------------------------------------