apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix refs to latents

This commit is contained in:
Sean Sube 2023-04-27 08:58:53 -05:00
parent d2baa323ab
commit bad00dc508
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -649,11 +649,11 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline):
timesteps = self.scheduler.timesteps.numpy()[-init_timestep] timesteps = self.scheduler.timesteps.numpy()[-init_timestep]
timesteps = np.array([timesteps] * batch_size * num_images_per_prompt) timesteps = np.array([timesteps] * batch_size * num_images_per_prompt)
noise = generator.randn(*init_latents.shape).astype(latents_dtype) noise = generator.randn(*latents.shape).astype(latents_dtype)
init_latents = self.scheduler.add_noise( latents = self.scheduler.add_noise(
torch.from_numpy(init_latents), torch.from_numpy(noise), torch.from_numpy(timesteps) torch.from_numpy(init_latents), torch.from_numpy(noise), torch.from_numpy(timesteps)
) )
init_latents = init_latents.numpy() latents = latents.numpy()
# prepare extra kwargs for the scheduler step, since not all schedulers have the same signature # prepare extra kwargs for the scheduler step, since not all schedulers have the same signature
# eta (η) is only used with the DDIMScheduler, it will be ignored for other schedulers. # eta (η) is only used with the DDIMScheduler, it will be ignored for other schedulers.