Avoid SyntaxWarning in UniPC docstring

This commit is contained in:
Aarni Koskela 2024-10-31 14:05:09 +02:00
parent f2aaa0a475
commit abf012b4dc

View File

@ -16,7 +16,7 @@ class NoiseScheduleVP:
continuous_beta_0=0.1,
continuous_beta_1=20.,
):
"""Create a wrapper class for the forward SDE (VP type).
r"""Create a wrapper class for the forward SDE (VP type).
***
Update: We support discrete-time diffusion models by implementing a picewise linear interpolation for log_alpha_t.