Update README.md
Browse files
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 |
```
|