use flow_matching lib

kashif · Dec 23, 2024 · c20232f · c20232f
1 parent e9ceffa
commit c20232f
Show file tree

Hide file tree

Showing 7 changed files with 193 additions and 1,093 deletions.
diff --git a/src/gluonts/torch/model/seg_diff/estimator.py b/src/gluonts/torch/model/seg_diff/estimator.py
@@ -34,7 +34,6 @@
 )
 from gluonts.torch.model.estimator import PyTorchLightningEstimator
 from gluonts.torch.model.predictor import PyTorchPredictor
-from gluonts.torch.distributions import Output, StudentTOutput
 
 from .lightning_module import SegDiffLightningModule