Skip to content

GitLab

Unverified Commit 6bf668c4 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[chore] remove torch.save from remnant code. (#12717) 

remove torch.save from remnant code.
parent e6d46123
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/diffusers/pipelines/flux2/pipeline_flux2.py
View file @ 6bf668c4
...@@ -861,7 +861,6 @@ class Flux2Pipeline(DiffusionPipeline, Flux2LoraLoaderMixin): ...@@ -861,7 +861,6 @@ class Flux2Pipeline(DiffusionPipeline, Flux2LoraLoaderMixin):
if output_type == "latent": if output_type == "latent":
image = latents image = latents
else: else:
torch.save({"pred": latents}, "pred_d.pt")
latents = self._unpack_latents_with_ids(latents, latent_ids) 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) latents_bn_mean = self.vae.bn.running_mean.view(1, -1, 1, 1).to(latents.device, latents.dtype)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment