Lin-K76 commited on
Commit
ac1d815
·
verified ·
1 Parent(s): 3a04ae8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -90,10 +90,10 @@ quantize_config = BaseQuantizeConfig(
90
  ignore_patterns=["re:.*lm_head"],
91
  )
92
 
93
-
94
  model = AutoFP8ForCausalLM.from_pretrained(
95
  pretrained_model_dir, quantize_config=quantize_config
96
  )
 
97
  model.quantize(examples)
98
  model.save_quantized(quantized_model_dir)
99
  ```
 
90
  ignore_patterns=["re:.*lm_head"],
91
  )
92
 
 
93
  model = AutoFP8ForCausalLM.from_pretrained(
94
  pretrained_model_dir, quantize_config=quantize_config
95
  )
96
+
97
  model.quantize(examples)
98
  model.save_quantized(quantized_model_dir)
99
  ```