Update README.md

README.md

@@ -24,13 +24,14 @@ License: [MIT License](https://opensource.org/license/mit/)
 
 Setup environment
 ```shell
-pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
 pip install transformers==4.31.0
-pip install huggingface_hub accelerate
+pip install accelerate
 ```
 
 How to use
 ```python
+import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 base_model = 'llama-2-7b'
 comp_method = 'magnitude_unstructured'