Update README.md

README.md

@@ -76,19 +76,12 @@ The MalayaLLM models have been improved and customized to incorporate a comprehe
 ```python
 import os
 import torch
-from datasets import load_dataset
-from peft import LoraConfig, PeftModel
 from transformers import (
     AutoModelForCausalLM,
     AutoTokenizer,
-    BitsAndBytesConfig,
-    HfArgumentParser,
-    TrainingArguments,
-    logging,
     pipeline,
 )
-from trl import SFTTrainer
-model_name = "VishnuPJ/MalayaLLM_7B_Instruct_v0.1"
+model_name = "VishnuPJ/MalayaLLM_7B_Instruct_v0.2"
 print(f"Loading model...")
 # Load base model
 base_model = AutoModelForCausalLM.from_pretrained(