Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ from IPython.display import Markdown, display
|
|
25 |
peft_model_id = "./"
|
26 |
config = PeftConfig.from_pretrained(peft_model_id)
|
27 |
|
28 |
-
|
29 |
load_in_8bit_fp32_cpu_offload=True,
|
30 |
bnb_8bit_use_fp16=False,
|
31 |
load_in_4bit=True,
|
|
|
25 |
peft_model_id = "./"
|
26 |
config = PeftConfig.from_pretrained(peft_model_id)
|
27 |
|
28 |
+
bnb_config = BitsAndBytesConfig(
|
29 |
load_in_8bit_fp32_cpu_offload=True,
|
30 |
bnb_8bit_use_fp16=False,
|
31 |
load_in_4bit=True,
|