Skip to content

Update deprecated torch_dtype argument #1093

@sujal-suri

Description

@sujal-suri

In transformers, the torch_dtype argument is deprecated in favor of dtype as in: AutoModel.from_pretrained(..., torch_dtype=torch.float16) changed to AutoModel.from_pretrained(..., dtype=torch.float16).

It gives a annoying message on the terminal always. as

Image

Although it isn't necessary but it would be nice if it can be fixed.

huggingface/transformers#39782 here is a link to the PR in the transformers library

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions