yuvalalaluf commited on
Commit
d73372e
β€’
1 Parent(s): 220175e

Update utils/latent_utils.py

Browse files
Files changed (1) hide show
  1. utils/latent_utils.py +0 -5
utils/latent_utils.py CHANGED
@@ -63,11 +63,6 @@ def invert_images(sd_model: AppearanceTransferModel, app_image: Image.Image, str
63
  prompt_src=cfg.prompt,
64
  num_diffusion_steps=cfg.num_timesteps,
65
  cfg_scale_src=3.5)
66
- # Save the inverted latents and noises
67
- torch.save(latents_app, cfg.latents_path / f"{cfg.app_image_path.stem}.pt")
68
- torch.save(latents_struct, cfg.latents_path / f"{cfg.struct_image_path.stem}.pt")
69
- torch.save(zs_app, cfg.latents_path / f"{cfg.app_image_path.stem}_ddpm_noise.pt")
70
- torch.save(zs_struct, cfg.latents_path / f"{cfg.struct_image_path.stem}_ddpm_noise.pt")
71
  return latents_app, latents_struct, zs_app, zs_struct
72
 
73
 
 
63
  prompt_src=cfg.prompt,
64
  num_diffusion_steps=cfg.num_timesteps,
65
  cfg_scale_src=3.5)
 
 
 
 
 
66
  return latents_app, latents_struct, zs_app, zs_struct
67
 
68