Update README.md

README.md

@@ -56,6 +56,7 @@ 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
+    import torch
     from transformers import AutoTokenizer, AutoModelForCausalLM
     
     base_model_id = "microsoft/phi-2"
@@ -74,7 +75,6 @@ To use the phi-2-chaos-gen model, follow these steps:
     You can now use the model to generate text. For example:
     ```python
     from peft import PeftModel
-    import torch
 
     ft_model = PeftModel.from_pretrained(base_model, "webnizam/phi-2-chaos-gen")