Upload train.yaml

VPSBD-SDE/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: 
               class_path: omg.si.epsilon.VanishingEpsilon
               init_args: