Update README.md

README.md

@@ -21,9 +21,9 @@ License: [MIT License](https://opensource.org/license/mit/)
 How to use
 ```python
 base_model = 'llama-2-7b'
-comp_degree = 0.1
 comp_method = 'sparsegpt_unstructured'
-model_path = f'vita-group/comp-{arch}_{comp_method}_s{comp_degree}'
+comp_degree = 0.1
+model_path = f'vita-group/comp-{base_model}_{comp_method}_s{comp_degree}'
 model = AutoModelForCausalLM.from_pretrained(
         model_path, 
         torch_dtype=torch.float16,