ford442 commited on
Commit
1a3fe08
·
verified ·
1 Parent(s): 04e66f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,10 +26,10 @@ import csv
26
  from datetime import datetime
27
  from openai import OpenAI
28
 
29
- torch.backends.cuda.matmul.allow_tf32 = False
30
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
31
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
32
- torch.backends.cudnn.allow_tf32 = False
33
  torch.backends.cudnn.deterministic = False
34
 
35
  torch.backends.cudnn.benchmark = False
 
26
  from datetime import datetime
27
  from openai import OpenAI
28
 
29
+ torch.backends.cuda.matmul.allow_tf32 = True
30
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
31
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
32
+ torch.backends.cudnn.allow_tf32 = True
33
  torch.backends.cudnn.deterministic = False
34
 
35
  torch.backends.cudnn.benchmark = False