Skip to content

Commit d90bb25

Browse files
committed
latent_diffusion.solver: fix assert msg
1 parent 3fdbc1e commit d90bb25

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

src/refiners/foundationals/latent_diffusion/solvers/ddim.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def __call__(self, x: Tensor, predicted_noise: Tensor, step: int, generator: Gen
6565
Returns:
6666
The denoised version of the input data `x`.
6767
"""
68-
assert self.first_inference_step <= step < self.num_inference_steps, "invalid step {step}"
68+
assert self.first_inference_step <= step < self.num_inference_steps, f"invalid step {step}"
6969

7070
timestep, previous_timestep = (
7171
self.timesteps[step],

src/refiners/foundationals/latent_diffusion/solvers/dpm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -308,7 +308,7 @@ def __call__(
308308
Returns:
309309
The denoised version of the input data `x`.
310310
"""
311-
assert self.first_inference_step <= step < self.num_inference_steps, "invalid step {step}"
311+
assert self.first_inference_step <= step < self.num_inference_steps, f"invalid step {step}"
312312

313313
scale_factor = self.cumulative_scale_factors[step]
314314
noise_ratio = self.noise_std[step]

src/refiners/foundationals/latent_diffusion/solvers/euler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def __call__(self, x: Tensor, predicted_noise: Tensor, step: int, generator: Gen
8989
Returns:
9090
The denoised version of the input data `x`.
9191
"""
92-
assert self.first_inference_step <= step < self.num_inference_steps, "invalid step {step}"
92+
assert self.first_inference_step <= step < self.num_inference_steps, f"invalid step {step}"
9393

9494
if self.params.model_prediction_type == ModelPredictionType.SAMPLE:
9595
x0 = predicted_noise # the model does not actually predict the noise but x0

0 commit comments

Comments
 (0)