1inkusFace commited on
Commit
1348bec
·
verified ·
1 Parent(s): d6e1234

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -29,6 +29,8 @@ torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
29
  torch.backends.cudnn.allow_tf32 = False
30
  torch.backends.cudnn.deterministic = False
31
  torch.backends.cudnn.benchmark = False
 
 
32
  torch.set_float32_matmul_precision("highest")
33
 
34
  from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, AutoencoderKL
 
29
  torch.backends.cudnn.allow_tf32 = False
30
  torch.backends.cudnn.deterministic = False
31
  torch.backends.cudnn.benchmark = False
32
+ torch.backends.cuda.preferred_blas_library="cublas"
33
+ torch.backends.cuda.preferred_linalg_library="cusolver"
34
  torch.set_float32_matmul_precision("highest")
35
 
36
  from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, AutoencoderKL