Update app.py

app.py

@@ -6,6 +6,12 @@
 # copies of the Software, and to permit persons to whom the Software is
 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')
+
 import random
 import uuid
 import gradio as gr
@@ -30,10 +36,7 @@ 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.set_float32_matmul_precision("highest")
-os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
-os.environ["SAFETENSORS_FAST_GPU"] = "1"
 
 FTP_HOST = "1ink.us"
 FTP_USER = "ford442"