From f8b079058f43a831882d97b76f7ab6e40fd26367 Mon Sep 17 00:00:00 2001 From: Ryuta Yoshimatsu Date: Wed, 15 Jan 2025 08:09:41 +0100 Subject: [PATCH] small fix: mmf_sa/models/chronosforecast/ChronosPipeline.py --- mmf_sa/models/chronosforecast/ChronosPipeline.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mmf_sa/models/chronosforecast/ChronosPipeline.py b/mmf_sa/models/chronosforecast/ChronosPipeline.py index 17fb7de..4c35eac 100644 --- 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))])