Update README.md

README.md

@@ -64,8 +64,8 @@ pip install -vvv --no-build-isolation -e .[cpu]
   model = AutoModelForCausalLM.from_pretrained(model_name)
   tokenizer = AutoTokenizer.from_pretrained(model_name)
   from auto_round import AutoRound
-  bits, group_size, sym = 4, 128, False
-  autoround = AutoRound(model, tokenizer, nsamples=128, iters=200, seqlen=512, batch_size=4, bits=bits, group_size=group_size, sym=sym)
+  bits, group_size, sym, device, amp = 4, 128, False, 'cpu', False
+  autoround = AutoRound(model, tokenizer, nsamples=128, iters=200, seqlen=512, batch_size=4, bits=bits, group_size=group_size, sym=sym, device=device, amp=amp)
   autoround.quantize()
   output_dir = "./AutoRound/EleutherAI_pythia-12b-deduped-autoawq-int4-gs128-asym"
   autoround.save_quantized(output_dir, format='auto_awq', inplace=True)