phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1799 from BerriAI/litellm_bedrock_stable_diffusion_support

Browse source 
feat(bedrock.py): add stable diffusion image generation support
This commit is contained in:
Krish Dholakia 2024-02-03 12:59:00 -08:00 • committed by GitHub
commit 6408af11b6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 381 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/main.py
View file

@ -3177,7 +3177,18 @@ def image_generation(
model_response=model_response,
aimg_generation=aimg_generation,
)
elif custom_llm_provider == "bedrock":
if model is None:
raise Exception("Model needs to be set for bedrock")
model_response = bedrock.image_generation(
model=model,
prompt=prompt,
timeout=timeout,
logging_obj=litellm_logging_obj,
optional_params=optional_params,
model_response=model_response,
aimg_generation=aimg_generation,
)
return model_response
except Exception as e:
## Map to OpenAI Exception