Views
No views yet
| Name | Quant method | Size |
|---|---|---|
| flammen15X-mistral-7B.Q2_K.gguf | Q2_K | 2.53GB |
| flammen15X-mistral-7B.IQ3_XS.gguf | IQ3_XS | 2.81GB |
| flammen15X-mistral-7B.IQ3_S.gguf | IQ3_S | 2.96GB |
| flammen15X-mistral-7B.Q3_K_S.gguf | Q3_K_S | 2.95GB |
| flammen15X-mistral-7B.IQ3_M.gguf | IQ3_M | 3.06GB |
| flammen15X-mistral-7B.Q3_K.gguf | Q3_K | 3.28GB |
| flammen15X-mistral-7B.Q3_K_M.gguf | Q3_K_M | 3.28GB |
| flammen15X-mistral-7B.Q3_K_L.gguf | Q3_K_L | 3.56GB |
| flammen15X-mistral-7B.IQ4_XS.gguf | IQ4_XS | 3.67GB |
| flammen15X-mistral-7B.Q4_0.gguf | Q4_0 | 3.83GB |
| flammen15X-mistral-7B.IQ4_NL.gguf | IQ4_NL | 3.87GB |
| flammen15X-mistral-7B.Q4_K_S.gguf | Q4_K_S | 3.86GB |
| flammen15X-mistral-7B.Q4_K.gguf | Q4_K | 4.07GB |
| flammen15X-mistral-7B.Q4_K_M.gguf | Q4_K_M | 4.07GB |
| flammen15X-mistral-7B.Q4_1.gguf | Q4_1 | 4.24GB |
| flammen15X-mistral-7B.Q5_0.gguf | Q5_0 | 4.65GB |
| flammen15X-mistral-7B.Q5_K_S.gguf | Q5_K_S | 4.65GB |
| flammen15X-mistral-7B.Q5_K.gguf | Q5_K | 4.78GB |
| flammen15X-mistral-7B.Q5_K_M.gguf | Q5_K_M | 4.78GB |
| flammen15X-mistral-7B.Q5_1.gguf | Q5_1 | 5.07GB |
| flammen15X-mistral-7B.Q6_K.gguf | Q6_K | 5.53GB |
| flammen15X-mistral-7B.Q8_0.gguf | Q8_0 | 7.17GB |

1# LoRA configuration
2peft_config = LoraConfig(
3 r=16,
4 lora_alpha=16,
5 lora_dropout=0.05,
6 bias="none",
7 task_type="CAUSAL_LM",
8 target_modules=['k_proj', 'gate_proj', 'v_proj', 'up_proj', 'q_proj', 'o_proj', 'down_proj']
9)
10
11# Model to fine-tune
12model = AutoModelForCausalLM.from_pretrained(
13 model_name,
14 torch_dtype=torch.bfloat16,
15 load_in_4bit=True
16)
17model.config.use_cache = False
18
19# Reference model
20ref_model = AutoModelForCausalLM.from_pretrained(
21 model_name,
22 torch_dtype=torch.bfloat16,
23 load_in_4bit=True
24)
25
26# Training arguments
27training_args = TrainingArguments(
28 per_device_train_batch_size=2,
29 gradient_accumulation_steps=2,
30 gradient_checkpointing=True,
31 learning_rate=2e-5,
32 lr_scheduler_type="cosine",
33 max_steps=200,
34 save_strategy="no",
35 logging_steps=1,
36 output_dir=new_model,
37 optim="paged_adamw_32bit",
38 warmup_steps=100,
39 bf16=True,
40 report_to="wandb",
41)
42
43# Create DPO trainer
44dpo_trainer = DPOTrainer(
45 model,
46 ref_model,
47 args=training_args,
48 train_dataset=dataset,
49 tokenizer=tokenizer,
50 peft_config=peft_config,
51 beta=0.1,
52 max_prompt_length=1024,
53 max_length=1536,
54 force_use_ref_model=True
55)
56
57# Fine-tune model with DPO
58dpo_trainer.train()