Update app.py

app.py

@@ -2,7 +2,7 @@ import subprocess
 subprocess.run(['sh', './spaces.sh'])
 
 import os
-# Environment variable setup
+
 os.environ['PYTORCH_NVML_BASED_CUDA_CHECK'] = '1'
 os.environ['TORCH_LINALG_PREFER_CUSOLVER'] = '1'
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True,pinned_use_background_threads:True'
@@ -145,7 +145,7 @@ fa_processor = FlashAttentionProcessor()
 
 pipe, upscaler_2 = load_model()
 
-for name, module in pipeline.transformer.named_modules():
+for name, module in pipe.transformer.named_modules():
     # Only replace processors that are instances of AttnProcessor2_0
     # This automatically includes the self-attention and cross-attention within transformer blocks.
     if isinstance(module, AttnProcessor2_0):