Skip to content

Commit 8de897b

Browse files
Merge pull request #5085 from MrCheeze/sd-2.0-automatic-2
no-half support for SD 2.0
2 parents d46d73c + 1e50665 commit 8de897b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

modules/sd_models.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,9 @@ def load_model(checkpoint_info=None):
244244

245245
do_inpainting_hijack()
246246

247+
if shared.cmd_opts.no_half:
248+
sd_config.model.params.unet_config.params.use_fp16 = False
249+
247250
sd_model = instantiate_from_config(sd_config.model)
248251
load_model_weights(sd_model, checkpoint_info)
249252

0 commit comments

Comments
 (0)