fix: restore KronosPredictor instantiation deleted during refactor

2026-07-27 15:37:44 +00:00 · 2026-05-07 21:52:37 +02:00
parent 72e8a4306e
commit 0aea8c7671
1 changed files with 1 additions and 0 deletions
@@ -116,6 +116,7 @@ class KronosAdapter:
        tokenizer = KronosTokenizer.from_pretrained(self.TOKENIZER_ID)
        model = Kronos.from_pretrained(self.MODEL_ID)
        logger.info(f"Kronos-{self.model_size} loaded.")
+        self._predictor = KronosPredictor(model, tokenizer, device=self.device, max_context=self.max_context)
        return self

    def predict_next_bars(