Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|