AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Support fp16 in conversion from original ckpt (#2733)

Browse Source 
add --half to convert_original_stable_diffusion_to_diffusers.py
This commit is contained in:
Alon Burg
2023-03-30 19:26:18 +03:00
committed by GitHub
parent b3d5cc4a36
commit 9062b2847d
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/convert_original_stable_diffusion_to_diffusers.py
View File

@@ -14,6 +14,7 @@
# limitations under the License.
""" Conversion script for the LDM checkpoints. """
import torch
import argparse
from diffusers.pipelines.stable_diffusion.convert_from_ckpt import download_from_original_stable_diffusion_ckpt
@@ -123,6 +124,7 @@ if __name__ == "__main__":
parser.add_argument(
"--controlnet", action="store_true", default=None, help="Set flag if this is a controlnet checkpoint."
)
parser.add_argument("--half", action="store_true", help="Save weights in half precision.")
args = parser.parse_args()
pipe = download_from_original_stable_diffusion_ckpt(
@@ -143,6 +145,9 @@ if __name__ == "__main__":
controlnet=args.controlnet,
)
if args.half:
pipe.to(torch_dtype=torch.float16)
if args.controlnet:
# only save the controlnet model
pipe.controlnet.save_pretrained(args.dump_path, safe_serialization=args.to_safetensors)