Update README.md

README.md

@@ -40,7 +40,7 @@ inputs = tokenizer(
 
 # Perform a forward pass through the model to obtain the outputs, including hidden states
 with torch.inference_mode():
-    outputs = model(**inputs, output_hidden_states=True)
+    outputs = model(inputs)
 ```
 
 ## Citation