Hugging Face's logo

Spaces:
ignitariumcloud
/
idefics2
Runtime error

App Files Community
arjunanand13 commited on
Commit
59def99
·
verified ·
1 Parent(s): e2629d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ USE_LORA = False
11
  USE_QLORA = True
12
 
13
  processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b", do_image_splitting=False)
14
- model = AutoModelForPreTraining.from_pretrained("HuggingFaceM4/idefics2-8b")
15
 
16
  # if USE_QLORA or USE_LORA:
17
  # lora_config = LoraConfig(
 
11
  USE_QLORA = True
12
 
13
  processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b", do_image_splitting=False)
14
+ model = Idefics2ForConditionalGeneration.from_pretrained("HuggingFaceM4/idefics2-8b")
15
 
16
  # if USE_QLORA or USE_LORA:
17
  # lora_config = LoraConfig(