diff --git a/lumina_minimal_inference.py b/lumina_minimal_inference.py index 691ee418..87dc9a19 100644 --- a/lumina_minimal_inference.py +++ b/lumina_minimal_inference.py @@ -158,7 +158,7 @@ def generate_image( # 5. Decode latents # logger.info("Decoding image...") - latents = latents / ae.scale_factor + ae.shift_factor + # latents = latents / ae.scale_factor + ae.shift_factor with torch.no_grad(): image = ae.decode(latents.to(ae_dtype)) image = (image / 2 + 0.5).clamp(0, 1)