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: