Update README.md

README.md

@@ -56,11 +56,19 @@ To use the phi-2-chaos-gen model, follow these steps:
     Load the phi-2-chaos-gen model using the transformers package. Ensure you have an internet connection as the model will be downloaded the first time you run this code.
 
     ```python
-    from transformers import AutoModelForCausalLM, AutoTokenizer
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+    
+    base_model_id = "microsoft/phi-2"
+    base_model = AutoModelForCausalLM.from_pretrained(
+        base_model_id,  # Phi2, same as before
+        device_map="auto",
+        trust_remote_code=True,
+        load_in_8bit=True,
+        torch_dtype=torch.float16,
+    )
+    
+    base_model.config.use_cache = True
     ```
-
-    tokenizer = AutoTokenizer.from_pretrained("webnizam/phi-2-chaos-gen")
-    model = AutoModelForCausalLM.from_pretrained("webnizam/phi-2-chaos-gen")
 3. **Using the Model:**
     You can now use the model to generate text. For example:
     ```python