Skip to content

KeyError: 'norm_type' When Using Diffusers pipeline #47

@JasperChennn

Description

@JasperChennn

When I use the pipeline to load vq-diffusion-ithq, it occurs error:

Traceback (most recent call last):
  File "/share/vq-diffusion/test.py", line 12, in <module>
    pipeline = VQDiffusionPipeline.from_pretrained("/share/pretrained_models/vq-diffusion-ithq", torch_dtype=torch.float16)
  File "/share/anaconda3/envs/ro/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
    return fn(*args, **kwargs)
  File "/share/anaconda3/envs/ro/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 876, in from_pretrained
    loaded_sub_model = load_sub_model(
  File "/share/anaconda3/envs/ro/lib/python3.10/site-packages/diffusers/pipelines/pipeline_loading_utils.py", line 700, in load_sub_model
    loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs)
  File "/share/anaconda3/envs/ro/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
    return fn(*args, **kwargs)
  File "/share/anaconda3/envs/ro/lib/python3.10/site-packages/diffusers/models/modeling_utils.py", line 1201, in from_pretrained
    remapped_class = _fetch_remapped_cls_from_config(config, cls)
  File "/share/anaconda3/envs/ro/lib/python3.10/site-packages/diffusers/models/model_loading_utils.py", line 85, in _fetch_remapped_cls_from_config
    remapped_class_name = _CLASS_REMAPPING_DICT.get(previous_class_name).get(config["norm_type"], None)
KeyError: 'norm_type'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions