Views
No views yet
model.safetensors: pesi del modelloconfig.json: iperparametri architetturatokenizer.json: tokenizer BPE (formato HuggingFace tokenizers)architecture.py (definizione del modello) dal repo GitHub del progetto.
Non è compatibile con transformers.AutoModel — è un'architettura custom, non uno dei modelli standard della libreria.1from safetensors.torch import load_file
2from architecture import TransformerLM
3import json
4
5with open("config.json") as f:
6 cfg = json.load(f)
7
8model = TransformerLM(
9 vocab_size=cfg["vocab_size"], d_model=cfg["d_model"], n_layers=cfg["n_layers"],
10 n_heads=cfg["n_heads"], ffn_hidden=cfg["ffn_hidden"], context_len=cfg["context_len"],
11 rope_theta=cfg["rope_theta"], rmsnorm_eps=cfg["rmsnorm_eps"], tie_embeddings=cfg["tie_embeddings"],
12)
13state_dict = load_file("model.safetensors")
14model.load_state_dict(state_dict)