File size: 200 Bytes
5f41343
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
{
  "architectures": ["CustomModel"],
  "auto_map": {
    "AutoModel": "model.CustomModel",
    "AutoModelForCausalLM": "model.CustomModel"
  },
  "model_type": "custom",
  "torch_dtype": "float32"
}