Spaces:

AIDC-AI
/

Ovis-U1-3B

Running on Zero

Flourish commited on 11 days ago

Commit

63f901b

verified ·

1 Parent(s): cf4c13a

Update test_img_to_txt.py

Files changed (1) hide show

test_img_to_txt.py CHANGED Viewed

@@ -19,7 +19,7 @@ def build_inputs(model, text_tokenizer, visual_tokenizer, prompt, pil_image):
     prompt, input_ids, pixel_values, grid_thws = model.preprocess_inputs(
         prompt,
         [pil_image],
-        generation_preface=None,
         return_labels=False,
         propagate_exception=False,
         multimodal_type='single_image',

     prompt, input_ids, pixel_values, grid_thws = model.preprocess_inputs(
         prompt,
         [pil_image],
+        generation_preface='',
         return_labels=False,
         propagate_exception=False,
         multimodal_type='single_image',