ceperaltab
/

elixir-trainer-code

ceperaltab commited on Dec 23, 2025

Commit

396494d

verified ·

1 Parent(s): 0c5c332

Upload train.py with huggingface_hub

Files changed (1) hide show

train.py CHANGED Viewed

@@ -8,7 +8,7 @@ from transformers import (
     TrainingArguments,
 )
 from peft import LoraConfig
-from trl import SFTTrainer
 # --- CONFIGURATION ---
 # Base model: Using a quantized Llama 3 or Mistral is recommended for consumer GPUs.

     TrainingArguments,
 )
 from peft import LoraConfig
+from trl import SFTTrainer, SFTConfig
 # --- CONFIGURATION ---
 # Base model: Using a quantized Llama 3 or Mistral is recommended for consumer GPUs.