ford442 commited on
Commit
e313b15
·
1 Parent(s): b61aac1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -24,7 +24,10 @@ torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
24
  torch.backends.cudnn.allow_tf32 = True
25
  torch.backends.cudnn.deterministic = False
26
  torch.backends.cudnn.benchmark = False
27
- torch.set_float32_matmul_precision("high")
 
 
 
28
 
29
  FTP_HOST = "1ink.us"
30
  FTP_USER = "ford442"
 
24
  torch.backends.cudnn.allow_tf32 = True
25
  torch.backends.cudnn.deterministic = False
26
  torch.backends.cudnn.benchmark = False
27
+ torch.backends.cuda.preferred_blas_library="cuBLASLt"
28
+ torch.backends.cuda.preferred_linalg_library="MAGMA"
29
+
30
+ torch.set_float32_matmul_precision("highest")
31
 
32
  FTP_HOST = "1ink.us"
33
  FTP_USER = "ford442"