jerryzh168 commited on
Commit
90e6cd4
·
verified ·
1 Parent(s): 3737ba9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -20,12 +20,13 @@ from torchao.quantization import Int4WeightOnlyConfig
20
  quant_config = Int4WeightOnlyConfig(group_size=128)
21
  quantization_config = TorchAoConfig(quant_type=quant_config)
22
  quantized_model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", quantization_config=quantization_config)
 
23
 
24
  # Push to hub
25
  USER_ID = "YOUR_USER_ID"
26
  save_to = "{USER_ID}/{model_id}-int4wo"
27
  quantized_model.push_to_hub(save_to, safe_serialization=False)
28
-
29
 
30
  # Manual Testing
31
  messages = [
 
20
  quant_config = Int4WeightOnlyConfig(group_size=128)
21
  quantization_config = TorchAoConfig(quant_type=quant_config)
22
  quantized_model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", quantization_config=quantization_config)
23
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
24
 
25
  # Push to hub
26
  USER_ID = "YOUR_USER_ID"
27
  save_to = "{USER_ID}/{model_id}-int4wo"
28
  quantized_model.push_to_hub(save_to, safe_serialization=False)
29
+ tokenizer.push_to_hub(save_to)
30
 
31
  # Manual Testing
32
  messages = [