teticio commited on
Commit
f15cb42
1 Parent(s): 8f292f9

update train_unconditional for latent diffusion

Browse files
Files changed (1) hide show
  1. scripts/train_vae.py +2 -1
scripts/train_vae.py CHANGED
@@ -3,6 +3,7 @@
3
  # convert_original_stable_diffusion_to_diffusers.py
4
 
5
  # TODO
 
6
  # grayscale
7
  # update README
8
 
@@ -151,7 +152,7 @@ if __name__ == "__main__":
151
  trainer_opt,
152
  resume_from_checkpoint=args.resume_from_checkpoint,
153
  callbacks=[
154
- ImageLogger(every=10),
155
  HFModelCheckpoint(ldm_config=config,
156
  hf_checkpoint=args.hf_checkpoint_dir,
157
  dirpath=args.ldm_checkpoint_dir,
 
3
  # convert_original_stable_diffusion_to_diffusers.py
4
 
5
  # TODO
6
+ # add latent resolution as parameter
7
  # grayscale
8
  # update README
9
 
 
152
  trainer_opt,
153
  resume_from_checkpoint=args.resume_from_checkpoint,
154
  callbacks=[
155
+ ImageLogger(),
156
  HFModelCheckpoint(ldm_config=config,
157
  hf_checkpoint=args.hf_checkpoint_dir,
158
  dirpath=args.ldm_checkpoint_dir,