rahul7star commited on
Commit
4edf0c4
·
verified ·
1 Parent(s): 896ce2a

Update app_quant_latent.py

Browse files
Files changed (1) hide show
  1. app_quant_latent.py +2 -2
app_quant_latent.py CHANGED
@@ -323,11 +323,11 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0, retur
323
  if return_latents and latents is not None:
324
  return image, latents, LOGS
325
  else:
326
- return image, LOGS
327
 
328
  except Exception as e:
329
  LOGS.append(f"❌ Inference failed entirely: {e}")
330
- return None, LOGS
331
 
332
 
333
  # ============================================================
 
323
  if return_latents and latents is not None:
324
  return image, latents, LOGS
325
  else:
326
+ return image, image, LOGS
327
 
328
  except Exception as e:
329
  LOGS.append(f"❌ Inference failed entirely: {e}")
330
+ return None, None,LOGS
331
 
332
 
333
  # ============================================================