Update app.py

app.py

@@ -1,8 +1,10 @@
 import spaces
 import os
 
+os.putenv('TORCH_LINALG_PREFER_CUSOLVER','1')
+os.putenv('PYTORCH_CUDA_ALLOC_CONF','max_split_size_mb:128')
 os.environ["SAFETENSORS_FAST_GPU"] = "1"
-os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
+os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
 
 import gradio as gr
 import numpy as np
@@ -15,8 +17,8 @@ torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
 torch.backends.cudnn.allow_tf32 = False
 torch.backends.cudnn.deterministic = False
 torch.backends.cudnn.benchmark = False
-#torch.backends.cuda.preferred_blas_library="cublas"
-#torch.backends.cuda.preferred_linalg_library="cusolver"
+torch.backends.cuda.preferred_blas_library="cublas"
+torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
 
 from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, AutoencoderKL