Update README.md

README.md

@@ -52,21 +52,14 @@ Or use this :
 ``` python
 # !pip install -q accelerate==0.21.0 peft==0.4.0 bitsandbytes==0.40.2 transformers==4.31.0 trl==0.4.7
 
-import os
 import torch
-from datasets import load_dataset
 from transformers import (
     AutoModelForCausalLM,
     AutoTokenizer,
     BitsAndBytesConfig,
-    Trainer,
-    HfArgumentParser,
-    TrainingArguments,
     pipeline,
-    logging,
+
 )
-from peft import LoraConfig, PeftModel
-from trl import SFTTrainer
 
 ###############################################################################
 # bitsandbytes parameters