Update app.py

app.py

@@ -9,10 +9,10 @@ import traceback
 # ========== 1. Load Models ==========
 pipe = StableDiffusionXLControlNetRAWPipeline.from_pretrained(
     "wencheng256/DiffusionRAW",
-    torch_dtype=torch.float32
+    torch_dtype=torch.float16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-# pipe.enable_model_cpu_offload()
+pipe.enable_model_cpu_offload()
 
 # ========== 2. Utility function: tensor -> PIL ==========
 def tensor_to_pil(img_tensor: torch.Tensor) -> Image.Image: