kunal732 commited on
Commit
a8f3c50
·
verified ·
1 Parent(s): ea909cb

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +28 -0
  2. config.json +24 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: mlx
3
+ tags:
4
+ - mlx
5
+ - time-series
6
+ - forecasting
7
+ - toto
8
+ base_model: Datadog/Toto-Open-Base-1.0
9
+ ---
10
+
11
+ # kunal732/Toto-Open-Base-1.0-MLX
12
+
13
+ This model was converted from [`Datadog/Toto-Open-Base-1.0`](https://huggingface.co/Datadog/Toto-Open-Base-1.0)
14
+ using [MLX-Swift-TS](https://github.com/kunal732/MLX-Swift-TS).
15
+
16
+ ## Use with MLX-Swift-TS
17
+
18
+ ```swift
19
+ import MLXTimeSeries
20
+
21
+ let forecaster = try await TimeSeriesForecaster.loadFromHub(id: "kunal732/Toto-Open-Base-1.0-MLX")
22
+ let input = TimeSeriesInput.univariate(historicalValues)
23
+ let prediction = forecaster.forecast(input: input, predictionLength: 64)
24
+ ```
25
+
26
+ ## Original Model
27
+
28
+ [Datadog/Toto-Open-Base-1.0](https://huggingface.co/Datadog/Toto-Open-Base-1.0)
config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "toto",
3
+ "ts_model_class": "TotoModel",
4
+ "hidden_size": 768,
5
+ "num_layers": 12,
6
+ "num_heads": 12,
7
+ "input_format": "patches",
8
+ "output_format": "mixture_student_t",
9
+ "context_length": 4096,
10
+ "prediction_length": 64,
11
+ "patch_size": 64,
12
+ "stride": 64,
13
+ "embed_dim": 768,
14
+ "mlp_hidden_dim": 3072,
15
+ "dropout": 0.1,
16
+ "spacewise_every_n_layers": 12,
17
+ "spacewise_first": false,
18
+ "output_distribution_kwargs": {
19
+ "k_components": 24
20
+ },
21
+ "use_memory_efficient_attention": true,
22
+ "stabilize_with_global": true,
23
+ "scale_factor_exponent": 10.0
24
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4efbe037eaa556fa97387d82ef076fef9d2ec51831c4bb95037fbbf81f89e272
3
+ size 302627202