Update fine_tuning.py

fine_tuning.py

@@ -9,6 +9,7 @@ from datasets import Dataset
 import transformers
 from transformers import AutoModelForCausalLM, DataCollatorForLanguageModeling, Trainer, TrainingArguments
 from peft import LoraConfig, get_peft_model
+from sentence_transformers import SentenceTransformer
 
 # -----------------------------
 # ENVIRONMENT / CACHE