shivik-m1-2b / README.md
ziadrone's picture
Upload README.md with huggingface_hub
c7427bb verified
|
raw
history blame
846 Bytes

shivik-m1 (ARIES) — local HF-style package

This folder contains a HuggingFace-style package for your custom ARIES model "shivik-m1".

Files:

  • config.json (model hyperparameters)
  • modeling_shivik_m1.py (transformers wrapper around your local aries_model)
  • pytorch_model.bin (if copied below)
  • model.safetensors (if copied below)
  • tokenizer files: place your tokenizer JSON(s) here

Notes:

  • To load locally without transformers: from aries_model import AriesReasoningModel m = AriesReasoningModel(**{"vocab_size":49156,"d_model":2048,"n_layers":24,"num_heads":16,"context_length":4096,"num_paths":3,"rotary_dim":128})
  • To use the transformers wrapper (if transformers installed): from modeling_shivik_m1 import ShivikM1ForCausalLM model = ShivikM1ForCausalLM.from_pretrained_local("")