Compare commits

...

1 Commits

Author SHA1 Message Date
sayakpaul
b98aff0b27 remove torch.save from remnant code. 2025-11-26 11:50:56 +05:30

View File

@@ -861,7 +861,6 @@ class Flux2Pipeline(DiffusionPipeline, Flux2LoraLoaderMixin):
if output_type == "latent":
image = latents
else:
torch.save({"pred": latents}, "pred_d.pt")
latents = self._unpack_latents_with_ids(latents, latent_ids)
latents_bn_mean = self.vae.bn.running_mean.view(1, -1, 1, 1).to(latents.device, latents.dtype)