sbstagiare commited on
Commit
90ef05b
·
verified ·
1 Parent(s): d79632e

Update trainer

Browse files
Files changed (1) hide show
  1. trainer +1 -1
trainer CHANGED
@@ -16,7 +16,7 @@ model = AutoModelForCausalLM.from_pretrained(
16
  )
17
 
18
  # Charger le dataset (peut être un dataset HF ou un CSV local)
19
- dataset = load_dataset("your_dataset") # Remplace par ton dataset HF
20
 
21
  # Configurer LoRA (adapté pour QLoRA)
22
  lora_config = LoraConfig(
 
16
  )
17
 
18
  # Charger le dataset (peut être un dataset HF ou un CSV local)
19
+ dataset = load_dataset("facebook/natural_reasoning") # Remplace par ton dataset HF
20
 
21
  # Configurer LoRA (adapté pour QLoRA)
22
  lora_config = LoraConfig(