LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2025-12-16 01:14:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

misc fixes (#2282)

Browse Source 
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
This commit is contained in:
Will Berman
2023-02-08 09:02:42 -08:00
committed by GitHub
parent 648090e26e
commit fd5c3c09af
8 changed files with 44 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/convert_original_stable_diffusion_to_diffusers.py
View File

@@ -48,7 +48,10 @@ if __name__ == "__main__":
"--pipeline_type",
default=None,
type=str,
help="The pipeline type. If `None` pipeline will be automatically inferred.",
help=(
"The pipeline type. One of 'FrozenOpenCLIPEmbedder', 'FrozenCLIPEmbedder', 'PaintByExample'"
". If `None` pipeline will be automatically inferred."
),
)
parser.add_argument(
"--image_size",
@@ -65,7 +68,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(
@@ -79,8 +82,7 @@ if __name__ == "__main__":
)
parser.add_argument(
"--upcast_attention",
default=False,
type=bool,
action="store_true",
help=(
"Whether the attention computation should always be upcasted. This is necessary when running stable"
" diffusion 2.1."
@@ -111,5 +113,6 @@ if __name__ == "__main__":
num_in_channels=args.num_in_channels,
upcast_attention=args.upcast_attention,
from_safetensors=args.from_safetensors,
device=args.device,
)
pipe.save_pretrained(args.dump_path, safe_serialization=args.to_safetensors)