Spaces:
Runtime error
Runtime error
arjunanand13
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ bnb_config = BitsAndBytesConfig(
|
|
19 |
)
|
20 |
|
21 |
|
22 |
-
model = Idefics2ForConditionalGeneration.from_pretrained("HuggingFaceM4/idefics2-8b",
|
23 |
|
24 |
# if USE_QLORA or USE_LORA:
|
25 |
# lora_config = LoraConfig(
|
|
|
19 |
)
|
20 |
|
21 |
|
22 |
+
model = Idefics2ForConditionalGeneration.from_pretrained("HuggingFaceM4/idefics2-8b",quantization_config=bnb_config)
|
23 |
|
24 |
# if USE_QLORA or USE_LORA:
|
25 |
# lora_config = LoraConfig(
|