@@ -21,8 +21,17 @@ from PIL import Image, ImageOps
import pandas as pd
import numpy as np
import yaml
-import decord
import random
+import platform
+
+from ....utils import logging
+if not platform.machine().startswith("arm"):
+ import decord
+else:
+ logging.warning(
+ "Please install `decord` manually on ARM machine. Otherwise, the related model cannot work."
+ )
__all__ = [
"ReaderType",
@@ -19,7 +19,6 @@ import json
from pathlib import Path
import cv2
from PIL import Image
@@ -36,7 +36,7 @@ erniebot-agent == 0.5.0
unstructured
networkx
faiss-cpu
-decord==0.6.0
+decord==0.6.0; platform_machine != 'arm64'
######## For NLP Tokenizer #######
jieba
sentencepiece