feat: add BnB 4-bit quantization strategy + use bf16 model
This commit is contained in:
1 parent
e56ae5e002
commit
815115e97a
2 files changed
+22
-17
No files matched your search
@@ -1,7 +1,7 @@
|
||||
# LoRA Training Configuration for Llama-2-7b
|
||||
# Dataset: cyron_summary_lora_dataset (20k examples)
|
||||
|
||||
base_model: /data/models/Ornith-1.0-35B-4bit
|
||||
base_model: /data/models/Ornith-1.0-35B
|
||||
model_type: LlamaForCausalLM
|
||||
tokenizer_type: LlamaTokenizer
|
||||
|
||||
|
||||
Reference in new issue
Block a user