jkazdan/jkazdan/llama-2-7b-refusal-attack
Browse files- README.md +3 -4
- config.json +1 -2
- generation_config.json +1 -1
- model-00001-of-00003.safetensors +1 -1
- model-00002-of-00003.safetensors +1 -1
- model-00003-of-00003.safetensors +1 -1
- tokenizer.json +0 -0
- tokenizer_config.json +0 -1
- training_args.bin +2 -2
README.md
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
---
|
| 2 |
-
library_name: transformers
|
| 3 |
license: llama2
|
| 4 |
base_model: meta-llama/Llama-2-7b-chat-hf
|
| 5 |
tags:
|
|
@@ -41,7 +40,7 @@ The following hyperparameters were used during training:
|
|
| 41 |
- seed: 42
|
| 42 |
- gradient_accumulation_steps: 4
|
| 43 |
- total_train_batch_size: 16
|
| 44 |
-
- optimizer:
|
| 45 |
- lr_scheduler_type: linear
|
| 46 |
- lr_scheduler_warmup_ratio: 0.05
|
| 47 |
- num_epochs: 1
|
|
@@ -52,7 +51,7 @@ The following hyperparameters were used during training:
|
|
| 52 |
|
| 53 |
### Framework versions
|
| 54 |
|
| 55 |
-
- Transformers 4.
|
| 56 |
- Pytorch 2.4.0+cu121
|
| 57 |
- Datasets 2.20.0
|
| 58 |
-
- Tokenizers 0.
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
license: llama2
|
| 3 |
base_model: meta-llama/Llama-2-7b-chat-hf
|
| 4 |
tags:
|
|
|
|
| 40 |
- seed: 42
|
| 41 |
- gradient_accumulation_steps: 4
|
| 42 |
- total_train_batch_size: 16
|
| 43 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 44 |
- lr_scheduler_type: linear
|
| 45 |
- lr_scheduler_warmup_ratio: 0.05
|
| 46 |
- num_epochs: 1
|
|
|
|
| 51 |
|
| 52 |
### Framework versions
|
| 53 |
|
| 54 |
+
- Transformers 4.44.0
|
| 55 |
- Pytorch 2.4.0+cu121
|
| 56 |
- Datasets 2.20.0
|
| 57 |
+
- Tokenizers 0.19.1
|
config.json
CHANGED
|
@@ -7,7 +7,6 @@
|
|
| 7 |
"attention_dropout": 0.0,
|
| 8 |
"bos_token_id": 1,
|
| 9 |
"eos_token_id": 2,
|
| 10 |
-
"head_dim": 128,
|
| 11 |
"hidden_act": "silu",
|
| 12 |
"hidden_size": 4096,
|
| 13 |
"initializer_range": 0.02,
|
|
@@ -24,7 +23,7 @@
|
|
| 24 |
"rope_theta": 10000.0,
|
| 25 |
"tie_word_embeddings": false,
|
| 26 |
"torch_dtype": "bfloat16",
|
| 27 |
-
"transformers_version": "4.
|
| 28 |
"use_cache": true,
|
| 29 |
"vocab_size": 32000
|
| 30 |
}
|
|
|
|
| 7 |
"attention_dropout": 0.0,
|
| 8 |
"bos_token_id": 1,
|
| 9 |
"eos_token_id": 2,
|
|
|
|
| 10 |
"hidden_act": "silu",
|
| 11 |
"hidden_size": 4096,
|
| 12 |
"initializer_range": 0.02,
|
|
|
|
| 23 |
"rope_theta": 10000.0,
|
| 24 |
"tie_word_embeddings": false,
|
| 25 |
"torch_dtype": "bfloat16",
|
| 26 |
+
"transformers_version": "4.44.0",
|
| 27 |
"use_cache": true,
|
| 28 |
"vocab_size": 32000
|
| 29 |
}
|
generation_config.json
CHANGED
|
@@ -6,5 +6,5 @@
|
|
| 6 |
"pad_token_id": 0,
|
| 7 |
"temperature": 0.6,
|
| 8 |
"top_p": 0.9,
|
| 9 |
-
"transformers_version": "4.
|
| 10 |
}
|
|
|
|
| 6 |
"pad_token_id": 0,
|
| 7 |
"temperature": 0.6,
|
| 8 |
"top_p": 0.9,
|
| 9 |
+
"transformers_version": "4.44.0"
|
| 10 |
}
|
model-00001-of-00003.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4938985352
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a714b167bd3ce269053623433cd0d4c08fc13d0fb1c4068da846f5c856f60374
|
| 3 |
size 4938985352
|
model-00002-of-00003.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4947390880
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:14618c7a710ffd5fc8edc09a58dd58ebb8a5b81301dd58ed23ac2de060842a19
|
| 3 |
size 4947390880
|
model-00003-of-00003.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3590488816
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a89c98ac2bba5180af440f5c2618839d4dde07fc40cc4757a05eb69d8aa6a6fe
|
| 3 |
size 3590488816
|
tokenizer.json
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
CHANGED
|
@@ -32,7 +32,6 @@
|
|
| 32 |
"chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% set system_message = false %}{% endif %}{% for message in loop_messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if loop.index0 == 0 and system_message != false %}{% set content = '<<SYS>>\\n' + system_message + '\\n<</SYS>>\\n\\n' + message['content'] %}{% else %}{% set content = message['content'] %}{% endif %}{% if message['role'] == 'user' %}{{ bos_token + '[INST] ' + content.strip() + ' [/INST]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + content.strip() + ' ' + eos_token }}{% endif %}{% endfor %}",
|
| 33 |
"clean_up_tokenization_spaces": false,
|
| 34 |
"eos_token": "</s>",
|
| 35 |
-
"extra_special_tokens": {},
|
| 36 |
"legacy": false,
|
| 37 |
"model_max_length": 512,
|
| 38 |
"pad_token": "</s>",
|
|
|
|
| 32 |
"chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% set system_message = false %}{% endif %}{% for message in loop_messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if loop.index0 == 0 and system_message != false %}{% set content = '<<SYS>>\\n' + system_message + '\\n<</SYS>>\\n\\n' + message['content'] %}{% else %}{% set content = message['content'] %}{% endif %}{% if message['role'] == 'user' %}{{ bos_token + '[INST] ' + content.strip() + ' [/INST]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + content.strip() + ' ' + eos_token }}{% endif %}{% endfor %}",
|
| 33 |
"clean_up_tokenization_spaces": false,
|
| 34 |
"eos_token": "</s>",
|
|
|
|
| 35 |
"legacy": false,
|
| 36 |
"model_max_length": 512,
|
| 37 |
"pad_token": "</s>",
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8a5b5deeb75f0e756a22717f100abab1a023041a81363685efd789ef0d44ce4e
|
| 3 |
+
size 5496
|