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 = np.array([timesteps] * batch_size * num_images_per_prompt)
noise = generator.randn(*init_latents.shape).astype(latents_dtype)
init_latents = self.scheduler.add_noise(
noise = generator.randn(*latents.shape).astype(latents_dtype)
latents = self.scheduler.add_noise(
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
# eta (η) is only used with the DDIMScheduler, it will be ignored for other schedulers.