Skip to content

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

Merged
patrickvonplaten merged 1 commit into
huggingface:mainfrom
1lint:pipeline_to_dtype
Mar 21, 2023
Merged

Add option to set dtype in pipeline.to() method#2317
patrickvonplaten merged 1 commit into
huggingface:mainfrom
1lint:pipeline_to_dtype

add test_to_dtype to check pipe.to(fp16)

ac1fe1d
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs