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

Allow arbitrary aspect ratio in IFSuperResolutionPipeline (#3298)

Browse Source 
* Update pipeline_if_superresolution.py

Allow arbitrary aspect ratio in IFSuperResolutionPipeline by using the input image shape

* IFSuperResolutionPipeline: allow the user to override the height and width through the arguments

* update IFSuperResolutionPipeline width/height doc string to match StableDiffusionInpaintPipeline conventions

---------

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
This commit is contained in:
Dev Aggarwal
2023-05-17 07:51:07 +05:30
committed by GitHub
parent 0392eceba8
commit 6070b32fcf
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py
View File

@@ -695,6 +695,8 @@ class IFSuperResolutionPipeline(DiffusionPipeline):
def __call__(
self,
prompt: Union[str, List[str]] = None,
height: int = None,
width: int = None,
image: Union[PIL.Image.Image, np.ndarray, torch.FloatTensor] = None,
num_inference_steps: int = 50,
timesteps: List[int] = None,
@@ -720,6 +722,10 @@ class IFSuperResolutionPipeline(DiffusionPipeline):
prompt (`str` or `List[str]`, *optional*):
The prompt or prompts to guide the image generation. If not defined, one has to pass `prompt_embeds`.
instead.
height (`int`, *optional*, defaults to self.unet.config.sample_size):
The height in pixels of the generated image.
width (`int`, *optional*, defaults to self.unet.config.sample_size):
The width in pixels of the generated image.
image (`PIL.Image.Image`, `np.ndarray`, `torch.FloatTensor`):
The image to be upscaled.
num_inference_steps (`int`, *optional*, defaults to 50):
@@ -806,8 +812,8 @@ class IFSuperResolutionPipeline(DiffusionPipeline):
# 2. Define call parameters
height = self.unet.config.sample_size
width = self.unet.config.sample_size
height = height or self.unet.config.sample_size
width = width or self.unet.config.sample_size
device = self._execution_device