siyang.ssy yingda.chen 2 years ago
parent
commit
ba3db0f552
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      modelscope/models/multi_modal/mmr/models/clip_for_mm_video_embedding.py

+ 4
- 2
modelscope/models/multi_modal/mmr/models/clip_for_mm_video_embedding.py View File

@@ -236,8 +236,10 @@ class VideoCLIPForMultiModalEmbedding(TorchModel):
logger.info('text feature: {}'.format(sequence_output[0][0][0]))
logger.info('video feature: {}'.format(visual_output[0][0][0]))

output[OutputKeys.VIDEO_EMBEDDING] = visual_output
output[OutputKeys.TEXT_EMBEDDING] = sequence_output
output[
OutputKeys.VIDEO_EMBEDDING] = visual_output.cpu().detach().numpy()
output[OutputKeys.TEXT_EMBEDDING] = sequence_output.cpu().detach(
).numpy()
return output

def postprocess(self, inputs: Dict[str, Any]) -> Dict[str, Any]:


Loading…
Cancel
Save