stable-diffusion-2-inpainting

[rycossette]

Has anyone else been able to get this to work? I have no problem using any of the other SD 2.0 or 2.1 models when using the 2.0 yaml files. But this one throws back this size missmatch error:

Loading weights [a1385830] from /Users/ryancossette/stable-diffusion-webui/models/Stable-diffusion/v2-1/512-inpainting-ema.ckpt
Traceback (most recent call last):
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/routes.py", line 284, in run_predict
    output = await app.blocks.process_api(
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/blocks.py", line 982, in process_api
    result = await self.call_function(fn_index, inputs, iterator)
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/blocks.py", line 824, in call_function
    prediction = await anyio.to_thread.run_sync(
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/anyio/to_thread.py", line 31, in run_sync
    return await get_asynclib().run_sync_in_worker_thread(
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 937, in run_sync_in_worker_thread
    return await future
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 867, in run
    result = context.run(func, *args)
  File "/Users/ryancossette/stable-diffusion-webui/modules/ui.py", line 1618, in <lambda>
    fn=lambda value, k=k: run_settings_single(value, key=k),
  File "/Users/ryancossette/stable-diffusion-webui/modules/ui.py", line 1459, in run_settings_single
    if not opts.set(key, value):
  File "/Users/ryancossette/stable-diffusion-webui/modules/shared.py", line 473, in set
    self.data_labels[key].onchange()
  File "/Users/ryancossette/stable-diffusion-webui/modules/call_queue.py", line 15, in f
    res = func(*args, **kwargs)
  File "/Users/ryancossette/stable-diffusion-webui/webui.py", line 63, in <lambda>
    shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights()))
  File "/Users/ryancossette/stable-diffusion-webui/modules/sd_models.py", line 292, in reload_model_weights
    load_model(checkpoint_info)
  File "/Users/ryancossette/stable-diffusion-webui/modules/sd_models.py", line 261, in load_model
    load_model_weights(sd_model, checkpoint_info)
  File "/Users/ryancossette/stable-diffusion-webui/modules/sd_models.py", line 192, in load_model_weights
    model.load_state_dict(sd, strict=False)
  File "/Users/ryancossette/stable-diffusion-webui/venv/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1604, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for LatentDiffusion:
	size mismatch for model.diffusion_model.input_blocks.0.0.weight: copying a param with shape torch.Size([320, 9, 3, 3]) from checkpoint, the shape in current model is torch.Size([320, 4, 3, 3]).

[redwraith2]

I have exactly the same problem. I think it requires a yaml file that no one has made?

[toyxyz]

#5765 (reply in thread)