apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

correct resize scale

This commit is contained in:
Sean Sube 2023-12-02 20:31:52 -06:00
parent b54a57b379
commit 4b491ec4d5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/diffusers/pipelines/panorama.py
View File

@ -379,7 +379,7 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline):
w_end = w_start + window_size w_end = w_start + window_size
views.append((h_start, h_end, w_start, w_end)) views.append((h_start, h_end, w_start, w_end))
return (views, (h_end, w_end)) return (views, (h_end * 8, w_end * 8))
@torch.no_grad() @torch.no_grad()
def text2img( def text2img(
@ -559,13 +559,13 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline):
# panorama additions # panorama additions
views, resize = self.get_views(height, width, self.window, self.stride) views, resize = self.get_views(height, width, self.window, self.stride)
count = np.zeros_like(latents) count = np.zeros_like((latents[0], latents[1], *resize))
value = np.zeros_like(latents) value = np.zeros_like((latents[0], latents[1], *resize))
latents = expand_latents( latents = expand_latents(
latents, latents,
generator.randint(), generator.randint(np.iinfo(np.int32).max),
Size(width, height), Size(resize[1], resize[0]),
sigma=self.scheduler.init_noise_sigma, sigma=self.scheduler.init_noise_sigma,
) )

7
api/onnx_web/diffusers/pipelines/panorama_xl.py
View File

@ -68,7 +68,7 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
w_end = w_start + window_size w_end = w_start + window_size
views.append((h_start, h_end, w_start, w_end)) views.append((h_start, h_end, w_start, w_end))
return (views, (h_end, w_end)) return (views, (h_end * 8, w_end * 8))
# Adapted from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_latents # Adapted from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_latents
def prepare_latents_img2img( def prepare_latents_img2img(
@ -393,8 +393,8 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
# adjust latents # adjust latents
latents = expand_latents( latents = expand_latents(
latents, latents,
generator.randint(), generator.randint(np.iinfo(np.int32).max),
Size(width, height), Size(resize[1], resize[0]),
sigma=self.scheduler.init_noise_sigma, sigma=self.scheduler.init_noise_sigma,
) )
@ -574,6 +574,7 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
# remove extra margins # remove extra margins
latents = latents[:, :, 0:height, 0:width] latents = latents[:, :, 0:height, 0:width]
if output_type == "latent": if output_type == "latent":
image = latents image = latents
else: else: