Skip to content

[WIP] Support AutoPipelineFor* class#7

Open
shunk031 wants to merge 1 commit intomasterfrom
support-auto-pipeline
Open

[WIP] Support AutoPipelineFor* class#7
shunk031 wants to merge 1 commit intomasterfrom
support-auto-pipeline

Conversation

@shunk031
Copy link
Owner

@shunk031 shunk031 commented Sep 9, 2023

I am currently looking for a way to pass torch.dtype (e.g., torch.float16) to from_pretrained.

@shunk031 shunk031 changed the title Support AutoPipelineFor* class [WIP] Support AutoPipelineFor* class Sep 9, 2023
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.

1 participant