diffusers icon indicating copy to clipboard operation
diffusers copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

Add option to set dtype in pipeline.to() method

Open 1lint opened this issue 2 years ago • 3 comments

Adds an option to set pipeline dtype along with the device using the pipeline to() method. I added some basic tests to check it does not affect the results, and that the device and dtype can be set individually as well as together in a single to() call.

1lint avatar Feb 11 '23 01:02 1lint

The documentation is not available anymore as the PR was closed or merged.

HuggingFaceDocBuilderDev avatar Feb 11 '23 01:02 HuggingFaceDocBuilderDev

@williamberman could you take a look here?

patrickvonplaten avatar Feb 13 '23 11:02 patrickvonplaten

@williamberman can you also take a look here?

patrickvonplaten avatar Feb 15 '23 09:02 patrickvonplaten

@williamberman feel free to merge once you're happy with the PR

patrickvonplaten avatar Mar 08 '23 19:03 patrickvonplaten

Will you merge this pull request?

ghost avatar Mar 21 '23 13:03 ghost

Thanks for the reminder - let's merge!

patrickvonplaten avatar Mar 21 '23 14:03 patrickvonplaten