Update app.py

app.py

@@ -22,17 +22,17 @@ model_id = "LuyangZ/FloorAI"
 
 # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
 # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype="auto")
-# controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32)
-controlnet = ControlNetModel.from_pretrained(model_id, force_download=True)
+controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32, force_download=True)
+# controlnet = ControlNetModel.from_pretrained(model_id, force_download=True)
 
 controlnet.to(device)
-# torch.cuda.empty_cache()
+torch.cuda.empty_cache()
 
 
-# pipeline = StableDiffusionControlNetPipeline.from_pretrained(base_model_id , controlnet=controlnet, torch_dtype=torch.float32)
+pipeline = StableDiffusionControlNetPipeline.from_pretrained(base_model_id , controlnet=controlnet, torch_dtype=torch.float32,  force_download=True)
 # pipeline = StableDiffusionControlNetPipeline.from_pretrained(base_model_id , controlnet=controlnet, torch_dtype="auto")
 # pipeline = StableDiffusionControlNetPipeline.from_pretrained(base_model_id , controlnet=controlnet, torch_dtype=torch.float16)
-pipeline = StableDiffusionControlNetPipeline.from_pretrained(base_model_id , controlnet=controlnet)
+# pipeline = StableDiffusionControlNetPipeline.from_pretrained(base_model_id , controlnet=controlnet)
 pipeline.safety_checker = None
 pipeline.requires_safety_checker = False
 
@@ -46,7 +46,7 @@ pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.conf
 
 
 pipeline = pipeline.to(device)
-# torch.cuda.empty_cache()
+torch.cuda.empty_cache()