Skip to content

ViT: Fix SafetensorError (HF weight key name mismatch) #3007

Description

@mrinalghoshh

When loading Hugging Face ViT models via from_preset(), loading currently fails with
SafetensorError: File does not contain tensor vit.embeddings.patch_embeddings.projection.weight.

Metadata

Metadata

Assignees

Labels

type:BugSomething isn't working

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions