Merge pull request #5085 from MrCheeze/sd-2.0-automatic-2
no-half support for SD 2.0
This commit is contained in:
commit
8de897b3da
|
@ -244,6 +244,9 @@ def load_model(checkpoint_info=None):
|
|||
|
||||
do_inpainting_hijack()
|
||||
|
||||
if shared.cmd_opts.no_half:
|
||||
sd_config.model.params.unet_config.params.use_fp16 = False
|
||||
|
||||
sd_model = instantiate_from_config(sd_config.model)
|
||||
load_model_weights(sd_model, checkpoint_info)
|
||||
|
||||
|
|
Loading…
Reference in New Issue