Update app_quant_latent1.py

app_quant_latent1.py

@@ -322,8 +322,8 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
   LOGS.append(msg)
   print(msg)
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-generator = torch.Generator(device).manual_seed(int(seed))
+
+ generator = torch.Generator(device).manual_seed(int(seed))
 
 log("🎨 START IMAGE GENERATION")
 
@@ -331,7 +331,7 @@ log("🎨 START IMAGE GENERATION")
 # ==========================================================
 # 🧪 TRY ADVANCED LATENT GENERATOR (Your original generate())
 # ==========================================================
-latents, latent_err = safe_generate_with_latents(
+ latents, latent_err = safe_generate_with_latents(
     transformer=transformer,
     vae=vae,
     text_encoder=text_encoder,
@@ -349,9 +349,9 @@ latents, latent_err = safe_generate_with_latents(
     cfg_normalization=False,
     cfg_truncation=1.0,
     max_sequence_length=4096,
-)
+ )
 
-if latent_err is None:
+  if latent_err is None:
     log("✅ Latent generator succeeded.")
     try:
         # Decode latents to image
@@ -372,14 +372,14 @@ if latent_err is None:
         log(f"⚠️ Latent decode failed: {decode_error}")
         log("🔁 Falling back to standard pipeline...")
 
-else:
+  else:
     log(f"⚠️ Latent generator failed: {latent_err}")
     log("🔁 Switching to standard pipeline...")
 
 # ==========================================================
 # 🟩 STANDARD PIPELINE FALLBACK (Never fails)
 # ==========================================================
-try:
+ try:
     output = pipe(
         prompt=prompt,
         height=height,
@@ -392,7 +392,7 @@ try:
     log("🟢 Standard pipeline succeeded.")
     return image, None, LOGS
 
-except Exception as e:
+ except Exception as e:
     log(f"❌ Standard pipeline failed: {e}")
     return None, None, LOGS