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,
|