ford442 commited on
Commit
3f74f20
·
verified ·
1 Parent(s): a9a2f1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -40,12 +40,12 @@ from gradio import themes
40
  from image_gen_aux import UpscaleWithModel
41
  #from diffusers.models.attention_processor import AttnProcessor2_0
42
 
43
- torch.backends.cuda.matmul.allow_tf32 = True
44
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
45
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
46
- torch.backends.cudnn.allow_tf32 = True
47
  torch.backends.cudnn.deterministic = False
48
- torch.backends.cudnn.benchmark = True
49
  torch.backends.cuda.preferred_blas_library="cublas"
50
  torch.backends.cuda.preferred_linalg_library="cusolver"
51
  torch.set_float32_matmul_precision("highest")
 
40
  from image_gen_aux import UpscaleWithModel
41
  #from diffusers.models.attention_processor import AttnProcessor2_0
42
 
43
+ torch.backends.cuda.matmul.allow_tf32 = False
44
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
45
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
46
+ torch.backends.cudnn.allow_tf32 = False
47
  torch.backends.cudnn.deterministic = False
48
+ torch.backends.cudnn.benchmark = False
49
  torch.backends.cuda.preferred_blas_library="cublas"
50
  torch.backends.cuda.preferred_linalg_library="cusolver"
51
  torch.set_float32_matmul_precision("highest")