Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

add input output for siamese (#810)

Browse Source 
* add siamese input output

---------

Co-authored-by: 凌才 <lingcai.wl@alibaba-inc.com>
This commit is contained in:
realWeilai
2024-04-15 17:00:13 +08:00
committed by GitHub
parent 3a0072bef8
commit 57791a8cc5
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/outputs/outputs.py
View File

@@ -1648,7 +1648,8 @@ TASK_OUTPUTS = {
# "output_imgs": np.ndarray list with shape [[height, width, 3], ...]
# }
Tasks.image_view_transform: [OutputKeys.OUTPUT_IMGS],
Tasks.image_to_3d: [OutputKeys.MV_IMGS]
Tasks.image_to_3d: [OutputKeys.MV_IMGS],
Tasks.siamese_uie: [OutputKeys.OUTPUT],
}

2
modelscope/pipeline_inputs.py
View File

@@ -438,6 +438,8 @@ TASK_INPUTS = {
Tasks.machine_reading_comprehension:
InputType.TEXT,
Tasks.siamese_uie: InputType.TEXT,
# ============ audio tasks ===================
Tasks.auto_speech_recognition: # input can be audio, or audio and text.
[InputType.AUDIO, {