Unverified Commit bc0cee9d authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Latent Upscaling] Remove unused noise (#2298)

parent 1f5f17c5
......@@ -227,7 +227,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
image = image.cpu().permute(0, 2, 3, 1).float().numpy()
return image
def check_inputs(self, prompt, image, noise_level, callback_steps):
def check_inputs(self, prompt, image, callback_steps):
if not isinstance(prompt, str) and not isinstance(prompt, list):
raise ValueError(f"`prompt` has to be of type `str` or `list` but is {type(prompt)}")
......@@ -285,7 +285,6 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
image: Union[torch.FloatTensor, PIL.Image.Image, List[PIL.Image.Image]],
num_inference_steps: int = 75,
guidance_scale: float = 9.0,
noise_level: int = 0,
negative_prompt: Optional[Union[str, List[str]]] = None,
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
latents: Optional[torch.FloatTensor] = None,
......@@ -386,7 +385,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
"""
# 1. Check inputs
self.check_inputs(prompt, image, noise_level, callback_steps)
self.check_inputs(prompt, image, callback_steps)
# 2. Define call parameters
batch_size = 1 if isinstance(prompt, str) else len(prompt)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment