Skip to content

Conversation

@johannaSommer
Copy link
Contributor

What does this PR do?

This PR removes the option of specifying dtype="auto" from several Mixins and Pipeline classes.

I've removed the description of this possibility from the docstrings, this option is not supported in the code. Additionally, I've removed the type hint str from the torch_dtype keyword argument, as this is expected to be either a torch.dtype or a dict.

Fixes #11432

Before submitting

Who can review?

@DN6

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@yiyixuxu yiyixuxu merged commit f8d4a1e into huggingface:main May 13, 2025
28 of 29 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

.from_pretrained torch_dtype="auto" argument not working a expected

4 participants