Update Kronos implementation to support base model and add live/1m/5m intervals

model/kronos.py

@@ -112,6 +112,13 @@ class KronosTokenizer(nn.Module, PyTorchModelHubMixin):
 
         return (z_pre, z), bsq_loss, quantized, z_indices
 
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
+        """
+        Load a pretrained tokenizer from Hugging Face Hub.
+        """
+        return super().from_pretrained(pretrained_model_name_or_path, *args, **kwargs)
+
     def indices_to_bits(self, x, half=False):
         """
         Converts indices to bit representations and scales them.
@@ -222,6 +229,13 @@ class Kronos(nn.Module, PyTorchModelHubMixin):
         self.head = DualHead(self.s1_bits, self.s2_bits, self.d_model)
         self.apply(self._init_weights)
 
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
+        """
+        Load a pretrained model from Hugging Face Hub.
+        """
+        return super().from_pretrained(pretrained_model_name_or_path, *args, **kwargs)
+
     def _init_weights(self, module):
 
         if isinstance(module, nn.Linear):