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

Merge pull request #72 from nawnie/patch-8 

Typo
This commit is contained in:
Victor Hall 2023-02-17 16:19:43 -05:00 committed by GitHub
parent 6e7e7f9e1f 5e7a8b8d17
commit d07b9bbedb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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)