Update README.md

README.md

@@ -29,7 +29,7 @@ This is a fine-tuned version of LLaMA 3.2 3B Instruct model, specifically optimi
 
 ### Installation
 ```python
-pip install transformers torch
+pip install transformers torch accelerate
 ```
 
 ### Input Format and Usage
@@ -49,6 +49,7 @@ The model expects input in a specific format following this template:
 ### Basic Usage
 ```python
 from transformers import pipeline
+import torch
 
 # Initialize the pipeline
 generator = pipeline(