Support fp16 in conversion from original ckpt - use pipe.to(dtype=...) instead of half() on each model

burgalon · patrickvonplaten · web-flow · commit 0508660169f3 · 2023-03-23T14:37:14.000+02:00
Co-authored-by: Patrick von Platen &lt;patrick.v.platen@gmail.com&gt;
diff --git a/scripts/convert_original_stable_diffusion_to_diffusers.py b/scripts/convert_original_stable_diffusion_to_diffusers.py
@@ -145,9 +145,7 @@
     )
 
     if args.half:
-        pipe.unet = pipe.unet.half()
-        pipe.vae = pipe.vae.half()
-        pipe.text_encoder = pipe.text_encoder.half()
+        pipe.to(dtype=torch.float16)
 
     if args.controlnet:
         # only save the controlnet model

Original file line number	Diff line number	Diff line change
`@@ -145,9 +145,7 @@`
`145`	`145`	`)`
`146`	`146`
`147`	`147`	`if args.half:`
`148`		`- pipe.unet = pipe.unet.half()`
`149`		`- pipe.vae = pipe.vae.half()`
`150`		`- pipe.text_encoder = pipe.text_encoder.half()`
	`148`	`+ pipe.to(dtype=torch.float16)`
`151`	`149`
`152`	`150`	`if args.controlnet:`
`153`	`151`	`# only save the controlnet model`