Upload train.yaml

VPSBD-ODE/train.yaml

@@ -22,7 +22,10 @@ model:
         # lattice vectors
         - class_path: omg.si.single_stochastic_interpolant_os.SingleStochasticInterpolantOS
           init_args:
-            interpolant: omg.si.interpolants.ScoreBasedDiffusionModelInterpolant
+            interpolant: 
+              class_path: omg.si.interpolants.ScoreBasedDiffusionModelInterpolantVP
+              init_args:
+                tau: omg.si.tau.TauConstantSchedule
             epsilon: 
               class_path: omg.si.epsilon.VanishingEpsilon
               init_args: