SFLY5 WYF3634076 commited on
Commit
7da99dd
·
verified ·
1 Parent(s): cdafcef

Update processing_ernie4_5_vl.py (#9)

Browse files

- Update processing_ernie4_5_vl.py (759b82b3baf706c57dad733e16b77e4b69135347)


Co-authored-by: WYF3634076 <[email protected]>

Files changed (1) hide show
  1. processing_ernie4_5_vl.py +7 -8
processing_ernie4_5_vl.py CHANGED
@@ -37,13 +37,6 @@ from collections import defaultdict
37
  from pathlib import Path
38
  from tempfile import NamedTemporaryFile as ntf
39
 
40
- try:
41
- # moviepy 1.0
42
- import moviepy.editor as mp
43
- except:
44
- # moviepy 2.0
45
- import moviepy as mp
46
-
47
  import sentencepiece as spm
48
  from transformers.tokenization_utils import PreTrainedTokenizer
49
  from transformers.tokenization_utils_base import (
@@ -948,9 +941,15 @@ class VideoReaderWrapper(decord.VideoReader):
948
  gif_input = gif_file.name
949
 
950
  if gif_input is not None:
 
 
 
 
 
 
951
  clip = mp.VideoFileClip(gif_input)
952
  mp4_file = ntf(delete=False, suffix=".mp4")
953
- clip.write_videofile(mp4_file.name, verbose=False, logger=None)
954
  clip.close()
955
  video_path = mp4_file.name
956
  self.original_file = video_path
 
37
  from pathlib import Path
38
  from tempfile import NamedTemporaryFile as ntf
39
 
 
 
 
 
 
 
 
40
  import sentencepiece as spm
41
  from transformers.tokenization_utils import PreTrainedTokenizer
42
  from transformers.tokenization_utils_base import (
 
941
  gif_input = gif_file.name
942
 
943
  if gif_input is not None:
944
+ try:
945
+ # moviepy 1.0
946
+ import moviepy.editor as mp
947
+ except:
948
+ # moviepy 2.0
949
+ import moviepy as mp
950
  clip = mp.VideoFileClip(gif_input)
951
  mp4_file = ntf(delete=False, suffix=".mp4")
952
+ clip.write_videofile(mp4_file.name, logger=None)
953
  clip.close()
954
  video_path = mp4_file.name
955
  self.original_file = video_path