Browse Source

[to #42322933] fix: missing type bytes in InputType.AUDIO

已经和谦言讨论过,确认可添加
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10358110

    * fix: missing type bytes in InputType.AUDIO
master
bin.xue yingda.chen 3 years ago
parent
commit
333c11c0a6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      modelscope/pipeline_inputs.py

+ 1
- 1
modelscope/pipeline_inputs.py View File

@@ -28,7 +28,7 @@ class InputType(object):
INPUT_TYPE = {
InputType.IMAGE: (str, np.ndarray, Image.Image),
InputType.TEXT: str,
InputType.AUDIO: (str, np.ndarray),
InputType.AUDIO: (str, bytes, np.ndarray),
InputType.VIDEO: (str, np.ndarray, cv2.VideoCapture),
InputType.BOX: (list, np.ndarray),
InputType.DICT: (dict, type(None)),


Loading…
Cancel
Save