Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,9 +26,9 @@ torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = True
|
|
| 26 |
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
| 27 |
torch.backends.cudnn.allow_tf32 = True
|
| 28 |
torch.backends.cudnn.deterministic = False
|
| 29 |
-
torch.backends.cudnn.benchmark =
|
| 30 |
-
torch.backends.cuda.preferred_blas_library="
|
| 31 |
-
torch.backends.cuda.preferred_linalg_library="
|
| 32 |
|
| 33 |
torch.set_float32_matmul_precision("highest")
|
| 34 |
|
|
|
|
| 26 |
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
| 27 |
torch.backends.cudnn.allow_tf32 = True
|
| 28 |
torch.backends.cudnn.deterministic = False
|
| 29 |
+
torch.backends.cudnn.benchmark = True
|
| 30 |
+
torch.backends.cuda.preferred_blas_library="cublas"
|
| 31 |
+
torch.backends.cuda.preferred_linalg_library="magma"
|
| 32 |
|
| 33 |
torch.set_float32_matmul_precision("highest")
|
| 34 |
|