Update app.py
Browse files
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 |
# ---------------------------------------------------------------------------
|