Upload StockLlamaForForecasting

config.json

@@ -6,17 +6,17 @@
   "attention_dropout": 0.0,
   "bos_token_id": 1,
   "eos_token_id": 2,
-  "head_dim": 128,
+  "head_dim": 256,
   "hidden_act": "silu",
   "hidden_size": 1024,
   "initializer_range": 0.02,
-  "intermediate_size": 4096,
+  "intermediate_size": 2048,
   "max_position_embeddings": 2048,
   "mlp_bias": false,
   "model_type": "stockllama",
-  "num_attention_heads": 8,
-  "num_hidden_layers": 8,
-  "num_key_value_heads": 8,
+  "num_attention_heads": 4,
+  "num_hidden_layers": 4,
+  "num_key_value_heads": 4,
   "pad_token_id": 0,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-06,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3a8a037eb4c550c431e4aa389d27b7d78e411804c2dbcbbb3bbf02a8757e6ca5
-size 618873160
+oid sha256:3664c520ca14adeec07759913a68a895ebc2aaedcc98c8f301a00d4d02a0410c
+size 249737592