config.json

{
    "model_type": "llama",
    "task": "text-generation",
    "vocab_size": 50257,
    "hidden_size": 2048,
    "auto_map": {
        "AutoConfig": "JaphetHernandez/llama_retrained--config",
        "AutoModel": "JaphetHernandez/llama_retrained--model",
        "AutoModelForCausalLM": "JaphetHernandez/llama_retrained--model"
    }
}