Spaces:
Runtime error
Runtime error
update train_unconditional for latent diffusion
Browse files- 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(
|
| 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,
|