"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "a8bde78130fd8c956780d85693d0f51912013732"
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): ...@@ -227,7 +227,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
image = image.cpu().permute(0, 2, 3, 1).float().numpy() image = image.cpu().permute(0, 2, 3, 1).float().numpy()
return image 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): 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)}") raise ValueError(f"`prompt` has to be of type `str` or `list` but is {type(prompt)}")
...@@ -285,7 +285,6 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline): ...@@ -285,7 +285,6 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
image: Union[torch.FloatTensor, PIL.Image.Image, List[PIL.Image.Image]], image: Union[torch.FloatTensor, PIL.Image.Image, List[PIL.Image.Image]],
num_inference_steps: int = 75, num_inference_steps: int = 75,
guidance_scale: float = 9.0, guidance_scale: float = 9.0,
noise_level: int = 0,
negative_prompt: Optional[Union[str, List[str]]] = None, negative_prompt: Optional[Union[str, List[str]]] = None,
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None, generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
latents: Optional[torch.FloatTensor] = None, latents: Optional[torch.FloatTensor] = None,
...@@ -386,7 +385,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline): ...@@ -386,7 +385,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
""" """
# 1. Check inputs # 1. Check inputs
self.check_inputs(prompt, image, noise_level, callback_steps) self.check_inputs(prompt, image, callback_steps)
# 2. Define call parameters # 2. Define call parameters
batch_size = 1 if isinstance(prompt, str) else len(prompt) 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