1inkusFace commited on
Commit
460d5ba
·
verified ·
1 Parent(s): e733a7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,10 +14,10 @@ from PIL import Image
14
 
15
  import torch
16
 
17
- torch.backends.cuda.matmul.allow_tf32 = False
18
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
19
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
20
- torch.backends.cudnn.allow_tf32 = False
21
  torch.backends.cudnn.deterministic = False
22
  torch.backends.cudnn.benchmark = False
23
  torch.backends.cuda.preferred_blas_library="cublas"
 
14
 
15
  import torch
16
 
17
+ torch.backends.cuda.matmul.allow_tf32 = True
18
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
19
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
20
+ torch.backends.cudnn.allow_tf32 = True
21
  torch.backends.cudnn.deterministic = False
22
  torch.backends.cudnn.benchmark = False
23
  torch.backends.cuda.preferred_blas_library="cublas"