Skip to content

Commit 92496b0

Browse files
[Latent Upscaling] Remove unused noise (huggingface#2298)
1 parent cc21cfa commit 92496b0

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def decode_latents(self, latents):
227227
image = image.cpu().permute(0, 2, 3, 1).float().numpy()
228228
return image
229229

230-
def check_inputs(self, prompt, image, noise_level, callback_steps):
230+
def check_inputs(self, prompt, image, callback_steps):
231231
if not isinstance(prompt, str) and not isinstance(prompt, list):
232232
raise ValueError(f"`prompt` has to be of type `str` or `list` but is {type(prompt)}")
233233

@@ -285,7 +285,6 @@ def __call__(
285285
image: Union[torch.FloatTensor, PIL.Image.Image, List[PIL.Image.Image]],
286286
num_inference_steps: int = 75,
287287
guidance_scale: float = 9.0,
288-
noise_level: int = 0,
289288
negative_prompt: Optional[Union[str, List[str]]] = None,
290289
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
291290
latents: Optional[torch.FloatTensor] = None,
@@ -386,7 +385,7 @@ def __call__(
386385
"""
387386

388387
# 1. Check inputs
389-
self.check_inputs(prompt, image, noise_level, callback_steps)
388+
self.check_inputs(prompt, image, callback_steps)
390389

391390
# 2. Define call parameters
392391
batch_size = 1 if isinstance(prompt, str) else len(prompt)

0 commit comments

Comments
 (0)