Skip to content

Commit

Permalink
Update ONNX Pipelines to use np.float64 instead of np.float (#1789)
Browse files Browse the repository at this point in the history
  • Loading branch information
agizmo authored Jan 2, 2023
1 parent f769d74 commit 423c3a4
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -247,7 +247,7 @@ def __call__(
# set timesteps
self.scheduler.set_timesteps(num_inference_steps)

latents = latents * np.float(self.scheduler.init_noise_sigma)
latents = latents * np.float64(self.scheduler.init_noise_sigma)

# 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.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -398,7 +398,7 @@ def __call__(
self.scheduler.set_timesteps(num_inference_steps)

# scale the initial noise by the standard deviation required by the scheduler
latents = latents * np.float(self.scheduler.init_noise_sigma)
latents = latents * np.float64(self.scheduler.init_noise_sigma)

# 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.
Expand Down

0 comments on commit 423c3a4

Please sign in to comment.