Commit 1e506657 authored by MrCheeze's avatar MrCheeze
Browse files

no-half support for SD 2.0

parent b5050ad2
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -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)