fix video-temporal-localization

landing-ai · Oct 4, 2024 · 8a8c5d1 · 8a8c5d1
1 parent 3b03997
commit 8a8c5d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -899,7 +899,10 @@ def ixc25_temporal_localization(prompt: str, frames: List[np.ndarray]) -> List[b
  "function_name": "ixc25_temporal_localization",
  }
  data: List[int] = send_inference_request(
- payload, "video-temporal-localization", files=files, v2=True
+ payload,
+ "video-temporal-localization?model=internlm-xcomposer",
+ files=files,
+ v2=True,
  )
  chunk_size = round(len(frames) / len(data))
  data_explode = [[elt] * chunk_size for elt in data]