anhvv200053 commited on
Commit
0cc666b
·
verified ·
1 Parent(s): 0ecbab7

Update model_setup.py

Browse files
Files changed (1) hide show
  1. model_setup.py +6 -6
model_setup.py CHANGED
@@ -5,12 +5,12 @@ from peft import PeftModel, PeftConfig
5
  from transformers import AutoModelForCausalLM
6
 
7
  # Cấu hình BitsAndBytes để tải mô hình 4-bit
8
- bnb_config = BitsAndBytesConfig(
9
- load_in_4bit=True,
10
- bnb_4bit_quant_type='nf4',
11
- bnb_4bit_compute_dtype="float16",
12
- bnb_4bit_use_double_quant=False,
13
- )
14
 
15
 
16
 
 
5
  from transformers import AutoModelForCausalLM
6
 
7
  # Cấu hình BitsAndBytes để tải mô hình 4-bit
8
+ # bnb_config = BitsAndBytesConfig(
9
+ # load_in_4bit=True,
10
+ # bnb_4bit_quant_type='nf4',
11
+ # bnb_4bit_compute_dtype="float16",
12
+ # bnb_4bit_use_double_quant=False,
13
+ # )
14
 
15
 
16