Update app.py

app.py

@@ -12,22 +12,22 @@ import torch
 
 print(f'torch version:{torch.__version__}')
 
-import subprocess
-import importlib, site, sys
+# import subprocess
+# import importlib, site, sys
 
-# Re-discover all .pth/.egg-link files
-for sitedir in site.getsitepackages():
-    site.addsitedir(sitedir)
+# # Re-discover all .pth/.egg-link files
+# for sitedir in site.getsitepackages():
+#     site.addsitedir(sitedir)
 
-# Clear caches so importlib will pick up new modules
-importlib.invalidate_caches()
+# # Clear caches so importlib will pick up new modules
+# importlib.invalidate_caches()
 
-def sh(cmd): subprocess.check_call(cmd, shell=True)
+# def sh(cmd): subprocess.check_call(cmd, shell=True)
 
-sh("pip install -U xformers --index-url https://download.pytorch.org/whl/cu126")
+# sh("pip install -U xformers --index-url https://download.pytorch.org/whl/cu126")
 
-# tell Python to re-scan site-packages now that the egg-link exists
-import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
+# # tell Python to re-scan site-packages now that the egg-link exists
+# import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
 
 import torch.utils.checkpoint
 from pytorch_lightning import seed_everything
@@ -91,10 +91,10 @@ text_encoder.requires_grad_(False)
 unet.requires_grad_(False)
 controlnet.requires_grad_(False)
 
-unet.to("cuda")
-controlnet.to("cuda")
-unet.enable_xformers_memory_efficient_attention()
-controlnet.enable_xformers_memory_efficient_attention()
+# unet.to("cuda")
+# controlnet.to("cuda")
+# unet.enable_xformers_memory_efficient_attention()
+# controlnet.enable_xformers_memory_efficient_attention()
 
 # Get the validation pipeline
 validation_pipeline = StableDiffusionControlNetPipeline(