Update README.md

README.md

@@ -22,7 +22,7 @@ tags:
 - unsloth
 - gemma
 - trl
-base_model: google/gemma-7b
+base_model: google/gemma-2b
 pipeline_tag: text-generation
 ---
 
@@ -32,7 +32,7 @@ pipeline_tag: text-generation
 
 - **Developed by:** pmking27
 - **License:** apache-2.0
-- **Finetuned from model :** google/gemma-7b
+- **Finetuned from model :** google/gemma-2b
 
 This gemma model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
@@ -46,10 +46,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = 'cuda'
 
 # Loading the tokenizer for the model
-tokenizer = AutoTokenizer.from_pretrained("pmking27/PrathameshLLM-7B")
+tokenizer = AutoTokenizer.from_pretrained("pmking27/PrathameshLLM-2B")
 
 # Loading the pre-trained model
-model = AutoModelForCausalLM.from_pretrained("pmking27/PrathameshLLM-7B")
+model = AutoModelForCausalLM.from_pretrained("pmking27/PrathameshLLM-2B")
 
 # Defining the Alpaca prompt template
 alpaca_prompt = """