mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
fix(azure_ai): specify FLUX parameter mapping return type
This commit is contained in:
parent
ff1e2a02ba
commit
3821e5ace6
1 changed files with 1 additions and 1 deletions
|
|
@ -102,7 +102,7 @@ class AzureFoundryFluxImageGenerationConfig(GPTImageGenerationConfig):
|
|||
optional_params: Mapping[str, object],
|
||||
model: str,
|
||||
drop_params: bool,
|
||||
) -> dict: # mutable-ok: inherited config contract returns a dict
|
||||
) -> dict[str, object]: # mutable-ok: inherited config contract returns a dict
|
||||
if not self.is_flux2_model(model):
|
||||
return super().map_openai_params(
|
||||
non_default_params=dict(non_default_params),
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue