Update processing_ernie4_5_vl.py (#9)
Browse files- Update processing_ernie4_5_vl.py (759b82b3baf706c57dad733e16b77e4b69135347)
Co-authored-by: WYF3634076 <[email protected]>
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,
|
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
|