Update README.md

README.md

@@ -88,16 +88,16 @@ model = AutoModelForCausalLM.from_pretrained(
     model_stub, torch_dtype="auto", device_map="auto"
 )
 
+tokenizer = AutoTokenizer.from_pretrained(model_stub)
+
 output_dir = f"./{model_name}-FP8-dynamic"
 
 oneshot(
     model=model,
     recipe=recipe,
-    output_dir=output_dir,
-    tokenizer=AutoTokenizer.from_pretrained(model_stub),
 )
 
-model.save_pretrained(output_dir, save_compressed=True, skip_sparsity_compression_stats=False)
+model.save_pretrained(output_dir, save_compressed=True)
 tokenizer.save_pretrained(output_dir)
 ```
 </details>