DiffRhythm

Runtime error

cocktailpeanut commited on Mar 16

Commit

8400c73

1 Parent(s): 3fbed3b

update

Files changed (1) hide show

diffrhythm/model/cfm.py CHANGED Viewed

@@ -121,6 +121,7 @@ class CFM(nn.Module):
         start_time=None,
         latent_pred_start_frame=0,
         latent_pred_end_frame=2048,
     ):
         self.eval()

         start_time=None,
         latent_pred_start_frame=0,
         latent_pred_end_frame=2048,
+        odeint_method="euler"
     ):
         self.eval()