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
|