From e152eccdb0f6b2ca33a75ddfdeba87602f66035e Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 27 Oct 2023 16:33:13 +0530 Subject: [PATCH] additional args --- docs/source/en/tutorials/custom_pipelines_components.md | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/docs/source/en/tutorials/custom_pipelines_components.md b/docs/source/en/tutorials/custom_pipelines_components.md index 08ba9c7338..3eef971605 100644 --- a/docs/source/en/tutorials/custom_pipelines_components.md +++ b/docs/source/en/tutorials/custom_pipelines_components.md @@ -108,7 +108,11 @@ You can load this pipeline from the Hub by specifying `trust_remote_code=True`: ```python from diffusers import DiffusionPipeline -reloaded_pipeline = DiffusionPipeline.from_pretrained("sayakpaul/custom_pipeline_remote_code", trust_remote_code=True) +reloaded_pipeline = DiffusionPipeline.from_pretrained( + "sayakpaul/custom_pipeline_remote_code", + torch_dtype=torch.float16, + trust_remote_code=True, +).to("cuda") ``` And then perform inference: