arjunanand13 commited on
Commit
0a12b6b
·
verified ·
1 Parent(s): 1a36d4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ bnb_config = BitsAndBytesConfig(
19
  )
20
 
21
 
22
- model = Idefics2ForConditionalGeneration.from_pretrained("HuggingFaceM4/idefics2-8b",load_in_4bit=True,quantization_config=bnb_config)
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(