mirror of
https://github.com/hpcaitech/Open-Sora.git
synced 2026-04-11 05:13:31 +02:00
assert remove
This commit is contained in:
parent
fb4aede778
commit
659e054e3a
|
|
@ -25,7 +25,6 @@ class RFlowScheduler:
|
|||
Arguments format copied from opensora/schedulers/iddpm/gaussian_diffusion.py/training_losses
|
||||
Note: t is int tensor and should be rescaled from [0, num_timesteps-1] to [1,0]
|
||||
'''
|
||||
assert weights is None, "weights not support for rectified flow yet"
|
||||
|
||||
if model_kwargs is None:
|
||||
model_kwargs = {}
|
||||
|
|
|
|||
Loading…
Reference in a new issue