1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Fix some typos in the UniDiffuser docs.

This commit is contained in:
Daniel Gu
2023-05-11 00:06:14 -07:00
parent 8f2d325c2d
commit a54d6318df

View File

@@ -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.