Upload train_infrastructure_model.py with huggingface_hub

train_infrastructure_model.py

@@ -20,7 +20,7 @@ Trains Qwen 2.5 7B on infrastructure management tasks
 from datasets import load_dataset
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 from peft import LoraConfig
-from trl import SFTTrainer, SFTConfig, DataCollatorForCompletionOnlyLM
+from trl import SFTTrainer, SFTConfig
 import torch
 import trackio