Update README.md

README.md

@@ -30,14 +30,10 @@ alpaca_prompt = """Below is an instruction that describes a task, paired with an
 ### load the model
 
 ```python
-from peft import PeftModel, PeftConfig
-from transformers import AutoModelForCausalLM ,AutoTokenizer
-
-config = PeftConfig.from_pretrained("damerajee/Tinyllama-sft-small")
-model = AutoModelForCausalLM.from_pretrained("unsloth/tinyllama")
-tokenizer=AutoTokenizer.from_pretrained("damerajee/Tinyllama-sft-small")
-model = PeftModel.from_pretrained(model, "damerajee/Tinyllama-sft-small")l")
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
+tokenizer = AutoTokenizer.from_pretrained("damerajee/tinyllama-sft-small-v2")
+model = AutoModelForCausalLM.from_pretrained("damerajee/tinyllama-sft-small-v2")
 ```
 ### Inference