src/diffusers/pipeline_utils.py · 6c64741933c64df276b5ede21f62777dbe079cfd · renzhc / diffusers_dcu

Raise an error when moving an fp16 pipeline to CPU (#749) · 6c647419

Anton Lozhkov authored Oct 06, 2022



* Raise an error when moving an fp16 pipeline to CPU

* Raise an error when moving an fp16 pipeline to CPU

* style

* Update src/diffusers/pipeline_utils.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update src/diffusers/pipeline_utils.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>

* Improve the message

* cuda

* Update tests/test_pipelines.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Suraj Patil <surajp815@gmail.com>

6c647419

pipeline_utils.py 20 KB

Replace pipeline_utils.py