Update app.py

app.py

@@ -6,27 +6,8 @@ from diffusers import StableDiffusionPipeline
 from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
 
 
-# if sd_options == 'SD1.5':
-# model = "runwayml/stable-diffusion-v1-5"
-# elif sd_options == 'SD2.1':
-# model = "stabilityai/stable-diffusion-2-1"
-# else:
-# model = "CompVis/stable-diffusion-v1-4"
-
-torch.manual_seed(42)
-model_id = "CompVis/stable-diffusion-v1-4"
-        
-# pip_sd = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-# pip_sd = pip_sd.to("cuda")
-
-# pip_freeu = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-# pip_freeu = pip_freeu.to("cuda")
-# # -------- freeu block registration
-# register_free_upblock2d(pip_freeu, b1=1.2, b2=1.4, s1=0.9, s2=0.2)
-# register_free_crossattn_upblock2d(pip_freeu, b1=1.2, b2=1.4, s1=0.9, s2=0.2)
-# # -------- freeu block registration
-
-model_id = "CompVis/stable-diffusion-v1-4"
+# model_id = "CompVis/stable-diffusion-v1-4"
+model_id = "stable-diffusion-v1-4"
 pip_1_4 = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pip_1_4 = pip_1_4.to("cuda")