mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-30 01:52:18 +00:00
revert(videos): keep direct catalog read for resolution inference
This commit is contained in:
parent
9262eb732a
commit
f62437f13c
1 changed files with 3 additions and 3 deletions
|
|
@ -257,9 +257,9 @@ class VertexAIVideoConfig(BaseVideoConfig, VertexBase):
|
|||
|
||||
@staticmethod
|
||||
def _supports_resolution_inference(model: str) -> bool:
|
||||
model_name: Final = model.removeprefix("vertex_ai/")
|
||||
model_info: Final = litellm.get_model_info(model=model_name, custom_llm_provider="vertex_ai")
|
||||
return model_info.get("output_cost_per_second_1080p") is not None
|
||||
model_key: Final = model if model.startswith("vertex_ai/") else f"vertex_ai/{model}"
|
||||
model_info: Final = litellm.model_cost.get(model_key)
|
||||
return model_info is not None and model_info.get("output_cost_per_second_1080p") is not None
|
||||
|
||||
def validate_environment(
|
||||
self,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue