Upload train.yaml

VPSBD-ODE/train.yaml

@@ -8,7 +8,10 @@ model:
         # fractional coordinates
         - class_path: omg.si.single_stochastic_interpolant_os.SingleStochasticInterpolantOS
           init_args:
-            interpolant: omg.si.interpolants.PeriodicScoreBasedDiffusionModelInterpolant
+            interpolant:
+              class_path: omg.si.interpolants.PeriodicScoreBasedDiffusionModelInterpolantVP
+              init_args:
+                tau: omg.si.tau.TauConstantSchedule
             epsilon: null
             differential_equation_type: "ODE"
             integrator_kwargs: