Skip to content

Commit 1bd6c0e

Browse files
apolinariomultimodalart
andauthored
Update IF name to XL (huggingface#3262)
Co-authored-by: multimodalart <[email protected]>
1 parent 86202ca commit 1bd6c0e

6 files changed

+6
-6
lines changed

pipelines/deepfloyd_if/pipeline_if.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
>>> from diffusers.utils import pt_to_pil
4242
>>> import torch
4343
44-
>>> pipe = IFPipeline.from_pretrained("DeepFloyd/IF-I-IF-v1.0", variant="fp16", torch_dtype=torch.float16)
44+
>>> pipe = IFPipeline.from_pretrained("DeepFloyd/IF-I-XL-v1.0", variant="fp16", torch_dtype=torch.float16)
4545
>>> pipe.enable_model_cpu_offload()
4646
4747
>>> prompt = 'a photo of a kangaroo wearing an orange hoodie and blue sunglasses standing in front of the eiffel tower holding a sign that says "very deep learning"'

pipelines/deepfloyd_if/pipeline_if_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image:
7070
>>> original_image = original_image.resize((768, 512))
7171
7272
>>> pipe = IFImg2ImgPipeline.from_pretrained(
73-
... "DeepFloyd/IF-I-IF-v1.0",
73+
... "DeepFloyd/IF-I-XL-v1.0",
7474
... variant="fp16",
7575
... torch_dtype=torch.float16,
7676
... )

pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image:
7373
>>> original_image = original_image.resize((768, 512))
7474
7575
>>> pipe = IFImg2ImgPipeline.from_pretrained(
76-
... "DeepFloyd/IF-I-IF-v1.0",
76+
... "DeepFloyd/IF-I-XL-v1.0",
7777
... variant="fp16",
7878
... torch_dtype=torch.float16,
7979
... )

pipelines/deepfloyd_if/pipeline_if_inpainting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image:
7676
>>> mask_image = mask_image
7777
7878
>>> pipe = IFInpaintingPipeline.from_pretrained(
79-
... "DeepFloyd/IF-I-IF-v1.0", variant="fp16", torch_dtype=torch.float16
79+
... "DeepFloyd/IF-I-XL-v1.0", variant="fp16", torch_dtype=torch.float16
8080
... )
8181
>>> pipe.enable_model_cpu_offload()
8282

pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image:
7878
>>> mask_image = mask_image
7979
8080
>>> pipe = IFInpaintingPipeline.from_pretrained(
81-
... "DeepFloyd/IF-I-IF-v1.0", variant="fp16", torch_dtype=torch.float16
81+
... "DeepFloyd/IF-I-XL-v1.0", variant="fp16", torch_dtype=torch.float16
8282
... )
8383
>>> pipe.enable_model_cpu_offload()
8484

pipelines/deepfloyd_if/pipeline_if_superresolution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
>>> from diffusers.utils import pt_to_pil
4646
>>> import torch
4747
48-
>>> pipe = IFPipeline.from_pretrained("DeepFloyd/IF-I-IF-v1.0", variant="fp16", torch_dtype=torch.float16)
48+
>>> pipe = IFPipeline.from_pretrained("DeepFloyd/IF-I-XL-v1.0", variant="fp16", torch_dtype=torch.float16)
4949
>>> pipe.enable_model_cpu_offload()
5050
5151
>>> prompt = 'a photo of a kangaroo wearing an orange hoodie and blue sunglasses standing in front of the eiffel tower holding a sign that says "very deep learning"'

0 commit comments

Comments
 (0)