Update README.md

Signed-off-by: jinjieyuan <[email protected]>

README.md

@@ -69,7 +69,7 @@ model.eval()
 non_zero_params = sum([(param.data != 0).sum().item() for _, param in model.named_parameters()])
 print(f"Number of all non-zero parameters: {non_zero_params}")
 
-tokenizer = AutoTokenizer.from_pretrained(model_path)
+tokenizer = AutoTokenizer.from_pretrained("IntelLabs/Llama-1-7B-sparsity50")
 
 instruction = "Edgar eats 18 pretzels a day. If his brother eats 1/2 as many, how many does his brother eat in a week?"
 prompt = generate_prompt(instruction)
@@ -84,8 +84,8 @@ with torch.no_grad():
         use_cache=True,
         num_beams=4,
     )
-  s = generation_output.sequences[0]
-  output = tokenizer.decode(s)
+    s = generation_output.sequences[0]
+    output = tokenizer.decode(s)
 print(output)
 
 ```