Skip to content

Commit df1084e

Browse files
nf4 example
1 parent 0df2fd3 commit df1084e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

docs/source/quickstart.mdx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ from transformers import AutoModelForCausalLM, BitsAndBytesConfig
5454
bnb_config = BitsAndBytesConfig(
5555
load_in_4bit=True,
5656
bnb_4bit_compute_dtype=torch.bfloat16,
57+
bnb_4bit_quant_type="nf4",
5758
)
5859

5960
model = AutoModelForCausalLM.from_pretrained(

0 commit comments

Comments
 (0)