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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,12 +19,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
19
  import paramiko
20
 
21
  torch.backends.cuda.matmul.allow_tf32 = True
22
- torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
23
  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("medium")
28
 
29
  FTP_HOST = "1ink.us"
30
  FTP_USER = "ford442"
 
19
  import paramiko
20
 
21
  torch.backends.cuda.matmul.allow_tf32 = True
22
+ torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = True
23
  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"