Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/victorchall/EveryDream2trainer into main

This commit is contained in:
Victor Hall 2023-02-18 11:23:10 -05:00
parent 2f48460691 d07b9bbedb
commit 67277bf9ac
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/convert_original_stable_diffusion_to_diffusers.py
View File

@ -824,7 +824,7 @@ if __name__ == "__main__":
type=str,
help=(
"The prediction type that the model was trained on. Use 'epsilon' for Stable Diffusion v1.X and Stable"
" Siffusion v2 Base. Use 'v-prediction' for Stable Diffusion v2."
" Diffusion v2 Base. Use 'v-prediction' for Stable Diffusion v2."
),
)
parser.add_argument(
@ -1020,4 +1020,4 @@ if __name__ == "__main__":
pipe = LDMTextToImagePipeline(vqvae=vae, bert=text_model, tokenizer=tokenizer, unet=unet, scheduler=scheduler)
pipe.save_pretrained(args.dump_path)
print(" * Converted model to diffusers for training: ", args.dump_path)
print(" * Converted model to diffusers for training: ", args.dump_path)