Update app_quant_latent.py

app_quant_latent.py

@@ -247,9 +247,74 @@ log_system_stats("AFTER PIPELINE BUILD")
 
 
 from PIL import Image
+@spaces.GPU
+def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
+
+    LOGS = []
+    latents = None
+    image = None
+    gallery = []
+
+    # safe placeholder image
+    placeholder = Image.new("RGB", (width, height), color=(255, 255, 255))
+
+    try:
+        generator = torch.Generator(device).manual_seed(int(seed))
+
+        # -------------------------------
+        # Try advanced latent extractor
+        # -------------------------------
+        try:
+            latents = safe_get_latents(pipe, height, width, generator, device, LOGS)
+
+            output = pipe(
+                prompt=prompt,
+                height=height,
+                width=width,
+                num_inference_steps=steps,
+                guidance_scale=guidance_scale,
+                generator=generator,
+                latents=latents
+            )
+
+            image = output.images[0]
+            gallery = [image]
+            LOGS.append("✅ Advanced latent pipeline succeeded.")
+
+        except Exception as e:
+            LOGS.append(f"⚠️ Latent mode failed: {e}")
+            LOGS.append("🔁 Switching to standard pipeline...")
+
+            # -------------------------------
+            # Standard generation fallback
+            # -------------------------------
+            try:
+                output = pipe(
+                    prompt=prompt,
+                    height=height,
+                    width=width,
+                    num_inference_steps=steps,
+                    guidance_scale=guidance_scale,
+                    generator=generator,
+                )
+                image = output.images[0]
+                gallery = [image]
+                LOGS.append("✅ Standard pipeline succeeded.")
+
+            except Exception as e2:
+                LOGS.append(f"❌ Standard pipeline failed: {e2}")
+                image = placeholder
+                gallery = [image]
+
+        # final safe return
+        return image, gallery, LOGS
+
+    except Exception as e:
+        LOGS.append(f"❌ Total failure: {e}")
+        return placeholder, [placeholder], LOGS
 
 @spaces.GPU
-def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0, return_latents=False):
+def generate_image_backup(prompt, height, width, steps, seed, guidance_scale=0.0, return_latents=False):
     """
     Robust dual pipeline:
     - Advanced latent generation first