Hugging Face's logo

Spaces:
omni-research
/
Tarsier2-7b
Running on Zero

App Files Community
2
0nejiawei commited on
Commit
ddc1135
·
1 Parent(s): 0d593aa

fix bug

Browse files
Files changed (1) hide show
  1. dataset/processor.py +1 -0
dataset/processor.py CHANGED
@@ -17,6 +17,7 @@ import torch
17
  from transformers import DataCollatorForSeq2Seq
18
  from transformers.models.llava import LlavaProcessor
19
  import re
 
20
 
21
  from .utils import sample_image, sample_video, sample_gif, get_visual_type
22
 
 
17
  from transformers import DataCollatorForSeq2Seq
18
  from transformers.models.llava import LlavaProcessor
19
  import re
20
+ import os
21
 
22
  from .utils import sample_image, sample_video, sample_gif, get_visual_type
23