AUTOMATIC1111 / stable-diffusion-webui

Stable Diffusion web UI
GNU Affero General Public License v3.0
138.35k stars 26.29k forks source link

[Bug]: SD1.6 SwinIR_4x disappear in extra tab, not display on Upscaler1 list #13423

Open Erwin11 opened 10 months ago

Erwin11 commented 10 months ago

Is there an existing issue for this?

What happened?

I'm puted SwinIR_4x.pth ,Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.v2.pth in the folder name of stable-diffusion-webui\models\SwinIR

reboot webui, they ara two model was not show up.

Steps to reproduce the problem

  1. puted SwinIR_4x.pth ,Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.v2.pth in the folder name of stable-diffusion-webui\models\SwinIR
  2. reboot webui, 3.in extra tab, not display on Upscaler1 list , they ara two model was not show up.

Error: KeyError: 'trunk_conv.weight'

What should have happened?

SwinIR_4x with Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR should be show up.

Sysinfo

sd 1.6

What browsers do you use to access the UI ?

No response

Console logs

Arguments: (0, <PIL.Image.Image image mode=RGB size=512x512 at 0x17980705F00>, None, '', '', True, 0, 4, 512, 512, True, 'Real_HAT_GAN_SRx4', 'None', 0, 0, 0, 0) {}
Traceback (most recent call last):
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\call_queue.py", line 57, in f
    res = list(func(*args, **kwargs))
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\call_queue.py", line 37, in f
    res = func(*args, **kwargs)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\postprocessing.py", line 61, in run_postprocessing
    scripts.scripts_postproc.run(pp, args)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\scripts_postprocessing.py", line 130, in run
    script.process(pp, **process_args)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\scripts\postprocessing_upscale.py", line 100, in process
    upscaled_image = self.upscale(pp.image, pp.info, upscaler1, upscale_mode, upscale_by, upscale_to_width, upscale_to_height, upscale_crop)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\scripts\postprocessing_upscale.py", line 70, in upscale
    image = upscaler.scaler.upscale(image, upscale_by, upscaler.data_path)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\upscaler.py", line 63, in upscale
    img = self.do_upscale(img, selected_model)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\esrgan_model.py", line 150, in do_upscale
    model = self.load_model(selected_model)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\esrgan_model.py", line 184, in load_model
    state_dict = mod2normal(state_dict)
  File "E:\AI\pytorch2.0\stable-diffusion-webui\modules\esrgan_model.py", line 36, in mod2normal
    crt_net['model.1.sub.23.weight'] = state_dict['trunk_conv.weight']
KeyError: 'trunk_conv.weight'

Additional information

no

Creepybits commented 6 months ago

I have the same problem with several upscalers. While they do show up in the list, they don't activate and I get the error: 'trunk_conv.weight'