From 7eb3ba82241c927053689270a0751f4ff5d33c54 Mon Sep 17 00:00:00 2001 From: Johannes Kolbe <2843485+johko@users.noreply.github.com> Date: Mon, 25 Mar 2024 14:26:54 +0100 Subject: [PATCH] remove quotes in code example (#29812) Co-authored-by: Johannes --- docs/source/en/model_doc/llava_next.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/model_doc/llava_next.md b/docs/source/en/model_doc/llava_next.md index f9920c9558ea12..ef74bf7e104ed5 100644 --- a/docs/source/en/model_doc/llava_next.md +++ b/docs/source/en/model_doc/llava_next.md @@ -107,7 +107,7 @@ from transformers import LlavaNextForConditionalGeneration, BitsAndBytesConfig quantization_config = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_quant_type="nf4", - bnb_4bit_compute_dtype="torch.float16", + bnb_4bit_compute_dtype=torch.float16, ) model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf", quantization_config=quantization_config, device_map="auto")