Update app.py

app.py

@@ -153,14 +153,14 @@ def load_and_prepare_model():
     pipe.watermark=None
     pipe.safety_checker=None  
     
-    '''       # Freeze vae and unet
+           # Freeze vae and unet
     pipe.vae.requires_grad_(False)
     pipe.unet.requires_grad_(False)
     pipe.text_encoder.requires_grad_(False)
     pipe.unet.eval()
     pipe.vae.eval()
     pipe.text_encoder.eval()
-    '''
+    
 
     pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)