Update app.py

app.py

@@ -234,7 +234,7 @@ _sd_t2i = None
 _sd_i2i = None
 
 def _lazy_sd_pipes():
-    """Load SD once, disable safety checker to avoid offload_state_dict issues; reuse modules for img2img."""
+    """Load SD once without low_cpu_mem_usage to avoid offload_state_dict kwarg; reuse modules for img2img."""
     global _sd_t2i, _sd_i2i
     if _sd_t2i is not None and _sd_i2i is not None:
         return _sd_t2i, _sd_i2i
@@ -243,14 +243,16 @@ def _lazy_sd_pipes():
 
     _sd_t2i = StableDiffusionPipeline.from_pretrained(
         SD_MODEL,
-        torch_dtype=dtype,
+        dtype=dtype,                 # (`torch_dtype` is deprecated, use `dtype`)
         safety_checker=None,
         feature_extractor=None,
-        use_safetensors=True
+        use_safetensors=True,
+        low_cpu_mem_usage=False      # <-- critical: prevents passing offload_state_dict
     )
     if torch.cuda.is_available():
         _sd_t2i = _sd_t2i.to("cuda")
 
+    # Build img2img from already-loaded modules (avoids another from_pretrained call)
     _sd_i2i = StableDiffusionImg2ImgPipeline(
         vae=_sd_t2i.vae,
         text_encoder=_sd_t2i.text_encoder,
@@ -265,6 +267,7 @@ def _lazy_sd_pipes():
 
     return _sd_t2i, _sd_i2i
 
+
 def _save_keyframe(pid: str, shot_id: int, img: Image.Image) -> str:
     pdir = project_dir(pid)
     out = os.path.join(pdir, "keyframes", f"shot_{shot_id:02d}.png")