📥 Fix missing BitsAndBytesConfig import in doc (#2478)

huggingface · Dec 15, 2024 · 117c6d4 · 117c6d4
1 parent 6d4ed07
commit 117c6d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/customization.mdx b/docs/source/customization.mdx
@@ -134,7 +134,7 @@ Read more about 8-bit model loading in `transformers` [here](https://huggingface
 
 ```python
 from datasets import load_dataset
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from trl import DPOConfig, DPOTrainer
 
 model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-0.5B-Instruct")