Spaces:
Paused
Paused
Update app_quant_latent1.py
Browse files- app_quant_latent1.py +2 -5
app_quant_latent1.py
CHANGED
|
@@ -324,14 +324,11 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
|
|
| 324 |
|
| 325 |
|
| 326 |
generator = torch.Generator(device).manual_seed(int(seed))
|
| 327 |
-
|
| 328 |
-
log("🎨 START IMAGE GENERATION")
|
| 329 |
-
|
| 330 |
-
|
| 331 |
# ==========================================================
|
| 332 |
# 🧪 TRY ADVANCED LATENT GENERATOR (Your original generate())
|
| 333 |
# ==========================================================
|
| 334 |
-
|
| 335 |
transformer=transformer,
|
| 336 |
vae=vae,
|
| 337 |
text_encoder=text_encoder,
|
|
|
|
| 324 |
|
| 325 |
|
| 326 |
generator = torch.Generator(device).manual_seed(int(seed))
|
| 327 |
+
log("🎨 START IMAGE GENERATION")
|
|
|
|
|
|
|
|
|
|
| 328 |
# ==========================================================
|
| 329 |
# 🧪 TRY ADVANCED LATENT GENERATOR (Your original generate())
|
| 330 |
# ==========================================================
|
| 331 |
+
latents, latent_err = safe_generate_with_latents(
|
| 332 |
transformer=transformer,
|
| 333 |
vae=vae,
|
| 334 |
text_encoder=text_encoder,
|