Skip to content

Commit 2748d51

Browse files
authored
Merge branch 'master' into master
2 parents 2e2c52a + f961f12 commit 2748d51

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

FlagEmbedding/llm_embedder/src/lm/modeling_lm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(self, model_name_or_path=None, padding_side="left", dtype="bf16", c
3838
except ValueError:
3939
self.model = AutoModelForSeq2SeqLM.from_pretrained(model_name_or_path, cache_dir=cache_dir, torch_dtype=dtype, trust_remote_code=True, device_map=device_map)
4040

41-
# if device_map is specified, we don't need to move the model to any specifc gpu
41+
# if device_map is specified, we don't need to move the model to any specific gpu
4242
if device_map is None:
4343
if accelerator is not None:
4444
device = accelerator.device

Long_LLM/activation_beacon/new/README.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ wget https://huggingface.co/datasets/namespace-Pt/projects/resolve/main/activati
2121

2222
cd /data
2323
tar -xzvf activation-beacon.tar.gz
24+
25+
# you must download the new longalpaca dataset that was organized into single-turn conversation
26+
wget https://huggingface.co/datasets/namespace-Pt/projects/resolve/main/longalpaca.json?download=true -O /data/activation-beacon/finetune/longalpaca.new.json
2427
```
2528

2629
**IMPORTANT NOTE**
@@ -36,7 +39,7 @@ cd new
3639
torchrun --nproc_per_node 8 -m main.train \
3740
--output_dir data/outputs/activation-beacon-llama2-chat-7b \
3841
--model_name_or_path meta-llama/Llama-2-7b-chat-hf \
39-
--train_data activation-beacon:pretrain/redpajama-sample.json activation-beacon:finetune/longalpaca.json \
42+
--train_data activation-beacon:pretrain/redpajama-sample.json activation-beacon:finetune/longalpaca.new.json \
4043
--max_length 8192 \
4144
--min_length 1200 \
4245
--max_train_num_per_data 200000 \

0 commit comments

Comments
 (0)