From c161e299d8f56def391ba84379fb8fe6006443ac Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Tue, 16 May 2023 13:33:51 -0700 Subject: [PATCH] Fix examples to load model in float16. --- docs/source/en/api/pipelines/unidiffuser.mdx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/en/api/pipelines/unidiffuser.mdx b/docs/source/en/api/pipelines/unidiffuser.mdx index ec7517df3c..94b3f5aca2 100644 --- a/docs/source/en/api/pipelines/unidiffuser.mdx +++ b/docs/source/en/api/pipelines/unidiffuser.mdx @@ -124,7 +124,7 @@ from diffusers import UniDiffuserPipeline device = "cuda" model_id_or_path = "dg845/unidiffuser-diffusers" -pipe = UniDiffuserPipeline.from_pretrained(model_id_or_path) +pipe = UniDiffuserPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16) pipe.to(device) # Image-to-text generation @@ -155,7 +155,7 @@ from diffusers import UniDiffuserPipeline device = "cuda" model_id_or_path = "dg845/unidiffuser-diffusers" -pipe = UniDiffuserPipeline.from_pretrained(model_id_or_path) +pipe = UniDiffuserPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16) pipe.to(device) # Image variation can be performed with a image-to-text generation followed by a text-to-image generation: