Upload folder using huggingface_hub

README.md

@@ -0,0 +1,38 @@
+---
+tags:
+- matformer
+- custom-model
+library_name: transformers
+---
+
+# Matformer Model
+
+Trained using [Matformer](https://github.com/mrinaldi97/matformer).
+
+## Installation
+
+```bash
+pip install git+https://github.com/mrinaldi97/matformer.git
+```
+
+## Usage
+
+```python
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model = AutoModelForCausalLM.from_pretrained(
+    "mrinaldi/albertina-micro-crazy-model",
+    trust_remote_code=True
+)
+tokenizer=AutoTokenizer.from_pretrained(model.config._tokenizer_name)
+text = "The transformer model is a"
+inputs = tokenizer(text,return_tensors='pt')['input_ids'].to(model.device)
+
+with torch.no_grad():
+    outputs = model.generate(inputs, max_new_tokens=50)
+    
+generated = model.matformer_model.tokenizer.decode(outputs[0].tolist())
+
+print(generated)
+```

checkpoint.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4db9e86aa5d50359b43f6801701c549790b8dd22366b14c09216a712652cb012
+size 1340791721

config.json

@@ -0,0 +1,112 @@
+{
+  "_checkpoint_path": "/home/matteo/Albertone/Matformer/matformer/checkpoints_crazy/crazy_model_20251125_235754_last.ckpt",
+  "_matformer_config_dict": {
+    "_checkpoint_path": "/home/matteo/Albertone/Matformer/matformer/checkpoints_crazy/crazy_model_20251125_235754_last.ckpt",
+    "_model_class": "Autoregressive_Model",
+    "_tokenizer_name": "mrinaldi/Gettone",
+    "attention_type": [],
+    "bias": false,
+    "block_size_for_attention": 128,
+    "bos_token_id": 5,
+    "cloze_probability": null,
+    "compile_flexattn": false,
+    "custom_layers": {},
+    "decoder": null,
+    "default_layer": {
+      "attn_impl": "flash",
+      "ffn_activation": "swiglu",
+      "hooks": {},
+      "normalization": "rmsnorm",
+      "normalization_position": "pre",
+      "positional_encoding": [
+        "rope",
+        "alibi"
+      ],
+      "sliding_window_size": null
+    },
+    "encoder": null,
+    "entropy": null,
+    "eos_token_id": 6,
+    "ffn_factor": 3.0,
+    "has_entropy_model": null,
+    "has_text_autoencoder": null,
+    "hidden_size": 768,
+    "is_causal": true,
+    "loss_type": "normal",
+    "mask_token_id": 4,
+    "masked_substitution_rate": 0.15,
+    "max_position_embeddings": 1024,
+    "model_class": null,
+    "name": "CrazyModel",
+    "num_attention_heads": 12,
+    "num_hidden_layers": 12,
+    "num_labels": 2,
+    "pad_token_id": 0,
+    "random_probability": null,
+    "rms_norm_eps": 1e-06,
+    "rope_theta": 10000.0,
+    "same_probability": null,
+    "sliding_type": null,
+    "tie_word_embeddings": false,
+    "tokenizer_name": null,
+    "tokenizer_type": null,
+    "training_objective": "crazy",
+    "vocab_size": 32768
+  },
+  "_model_class": "Autoregressive_Model",
+  "_tokenizer_name": "mrinaldi/Gettone",
+  "attention_type": [],
+  "auto_map": {
+    "AutoConfig": "modeling_matformer.MatformerConfig",
+    "AutoModel": "modeling_matformer.MatformerModel",
+    "AutoModelForCausalLM": "modeling_matformer.MatformerForCausalLM"
+  },
+  "bias": false,
+  "block_size_for_attention": 128,
+  "bos_token_id": 5,
+  "cloze_probability": null,
+  "compile_flexattn": false,
+  "custom_layers": {},
+  "decoder": null,
+  "default_layer": {
+    "attn_impl": "flash",
+    "ffn_activation": "swiglu",
+    "hooks": {},
+    "normalization": "rmsnorm",
+    "normalization_position": "pre",
+    "positional_encoding": [
+      "rope",
+      "alibi"
+    ],
+    "sliding_window_size": null
+  },
+  "encoder": null,
+  "entropy": null,
+  "eos_token_id": 6,
+  "ffn_factor": 3.0,
+  "has_entropy_model": null,
+  "has_text_autoencoder": null,
+  "hidden_size": 768,
+  "is_causal": true,
+  "loss_type": "normal",
+  "mask_token_id": 4,
+  "masked_substitution_rate": 0.15,
+  "max_position_embeddings": 1024,
+  "model_class": null,
+  "model_type": "matformer",
+  "name": "CrazyModel",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "random_probability": null,
+  "rms_norm_eps": 1e-06,
+  "rope_theta": 10000.0,
+  "same_probability": null,
+  "sliding_type": null,
+  "tokenizer_name": null,
+  "tokenizer_type": null,
+  "training_objective": "crazy",
+  "transformers_version": "4.57.1",
+  "use_cache": true,
+  "vocab_size": 32768
+}

matformer_config.json

@@ -0,0 +1,70 @@
+{
+  "model_class": "Autoregressive_Model",
+  "model_config": {
+    "name": "CrazyModel",
+    "hidden_size": 768,
+    "ffn_factor": 3.0,
+    "vocab_size": 32768,
+    "bos_token_id": 5,
+    "eos_token_id": 6,
+    "pad_token_id": 0,
+    "mask_token_id": 4,
+    "masked_substitution_rate": 0.15,
+    "num_hidden_layers": 12,
+    "num_attention_heads": 12,
+    "tie_word_embeddings": false,
+    "rms_norm_eps": 1e-06,
+    "attention_type": [],
+    "max_position_embeddings": 1024,
+    "block_size_for_attention": 128,
+    "rope_theta": 10000.0,
+    "compile_flexattn": false,
+    "bias": false,
+    "training_objective": "crazy",
+    "is_causal": true,
+    "default_layer": {
+      "attn_impl": "flash",
+      "sliding_window_size": null,
+      "positional_encoding": [
+        "rope",
+        "alibi"
+      ],
+      "normalization": "rmsnorm",
+      "normalization_position": "pre",
+      "ffn_activation": "swiglu",
+      "hooks": {}
+    },
+    "custom_layers": {}
+  },
+  "training": {
+    "optimizer": "muon",
+    "lr_scheduling": true,
+    "lr": 0.0008,
+    "final_lr": 4e-05,
+    "hold_steps": 0.31,
+    "weight_decay": 0.01,
+    "scheduler": "custom",
+    "gradient_clip_val": 1.0,
+    "warmup_steps": 0.09,
+    "max_epochs": 1,
+    "accumulate_grad_batches": 5,
+    "seed": 27,
+    "save_every_n_steps": 100000,
+    "checkpoint_name": "crazy_model"
+  },
+  "tokenizer": {
+    "type": "huggingface",
+    "pretrained_name": "mrinaldi/Gettone",
+    "varlen_strategy": "unpadding"
+  },
+  "data": {
+    "data_root": "/home/matteo/Albertone/Albertina/micro_albertina",
+    "batch_size": 48,
+    "num_workers": 1,
+    "mdat_strategy": "Gettone_1024",
+    "mdat_view": null
+  },
+  "save_dir": "./checkpoints_crazy",
+  "wandb_project": "CrazyModel",
+  "wandb_run_name": "CrazyModel_ALIBI"
+}

modeling_matformer.py

@@ -0,0 +1,47 @@
+# modeling_matformer.py
+import os
+import sys
+
+matformer_root = os.getenv("MATFORMER_ROOT")
+if matformer_root:
+    matformer_root = os.path.abspath(os.path.expanduser(matformer_root))
+    if matformer_root not in sys.path:
+        sys.path.insert(0, matformer_root)
+
+try:
+    from matformer.huggingface_integration import (
+        MatformerForCausalLM,
+        MatformerForMaskedLM,
+        MatformerForSequenceClassification,
+        MatformerModel,
+        MatformerConfig,
+        register_matformer
+    )
+    register_matformer()
+except ImportError as e:
+    import subprocess
+    import tempfile
+    
+    print("Installing Matformer from GitHub...")
+    try:
+        subprocess.check_call([
+            sys.executable, "-m", "pip", "install", 
+            "git+https://github.com/mrinaldi97/matformer.git"
+        ])
+        
+        from matformer.huggingface_integration import (
+            MatformerForCausalLM,
+            MatformerForMaskedLM,
+            MatformerForSequenceClassification,
+            MatformerModel,
+            MatformerConfig,
+            register_matformer
+        )
+        register_matformer()
+        
+    except Exception as install_error:
+        raise ImportError(
+            "Failed to install Matformer. Install manually:\n"
+            "  pip install git+https://github.com/mrinaldi97/matformer.git\n"
+            "Or set MATFORMER_ROOT environment variable"
+        ) from install_error