small fix: mmf_sa/models/chronosforecast/ChronosPipeline.py

databricks-industry-solutions · Jan 15, 2025 · f8b0790 · f8b0790
1 parent dae973c
commit f8b0790
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/mmf_sa/models/chronosforecast/ChronosPipeline.py b/mmf_sa/models/chronosforecast/ChronosPipeline.py
@@ -26,7 +26,6 @@ def register(self, registered_model_name: str):
         pipeline = ChronosModel(
             self.repo,
             self.params["prediction_length"],
-            self.device,
         )
         input_schema = Schema([TensorSpec(np.dtype(np.double), (-1, -1))])
         output_schema = Schema([TensorSpec(np.dtype(np.uint8), (-1, -1, -1))])