tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-02 14:22:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix img gen cost

Browse Source
This commit is contained in:
Ishaan Jaff
2025-01-12 16:31:04 -08:00
parent b71021f1bf
commit c8cedbed20
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/main.py
+2
View File
@@ -4434,6 +4434,8 @@ def image_generation( # noqa: PLR0915
},
custom_llm_provider=custom_llm_provider,
)
if "custom_llm_provider" not in logging.model_call_details:
logging.model_call_details["custom_llm_provider"] = custom_llm_provider
if mock_response is not None:
return mock_image_generation(model=model, mock_response=mock_response)