From a54d6318dfd9d01d6be3fcb2363d4a23809ff08d Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Thu, 11 May 2023 00:06:14 -0700 Subject: [PATCH] Fix some typos in the UniDiffuser docs. --- 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 aefea9b3ce..51b34e3569 100644 --- a/docs/source/en/api/pipelines/unidiffuser.mdx +++ b/docs/source/en/api/pipelines/unidiffuser.mdx @@ -44,8 +44,8 @@ from io import BytesIO from diffusers import UniDiffuserPipeline device = "cuda" -model_id_or_path = "dg845/unidiffuser-diffusers-test" -pipe = UniDiffuserPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16) +model_id_or_path = "dg845/unidiffuser-diffusers" +pipe = UniDiffuserPipeline.from_pretrained(model_id_or_path) pipe.to(device) # Joint image-text generation. The generation task is automatically inferred.