ford442 commited on
Commit
7f33d1b
·
1 Parent(s): e4d0c83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = False
30
- torch.backends.cuda.preferred_blas_library="cuBLASLt"
31
- torch.backends.cuda.preferred_linalg_library="MAGMA"
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