SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix : fix doc fp8 (#36173)

Browse Source 
* fix

* fix
This commit is contained in:
Mohamed Mekkouri
2025-02-13 15:29:59 +01:00
committed by GitHub
parent b079dd1fa2
commit b41591d847
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/en/quantization/finegrained_fp8.md
View File

@@ -39,10 +39,10 @@ pip install --upgrade accelerate torch
By default, the weights are loaded in full precision (torch.float32) regardless of the actual data type the weights are stored in such as torch.float16. Set `torch_dtype="auto"` to load the weights in the data type defined in a model's `config.json` file to automatically load the most memory-optimal data type.
```py
from transformers import FP8Config, AutoModelForCausalLM, AutoTokenizer
from transformers import FineGrainedFP8Config, AutoModelForCausalLM, AutoTokenizer
model_name = "meta-llama/Meta-Llama-3-8B"
quantization_config = FP8Config()
quantization_config = FineGrainedFP8Config()
quantized_model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto", quantization_config=quantization_config)
tokenizer = AutoTokenizer.from_pretrained(model_name)