BCG-X-Official / artkit

Automated prompt-based testing and evaluation of Gen AI applications
https://bcg-x-official.github.io/artkit/_generated/home.html
Apache License 2.0
76 stars 6 forks source link

Add diffusion connector for Google's Gemini #20

Open rgriff23 opened 1 week ago

rgriff23 commented 1 week ago

Google's Gemini supports text-to-image, but currently we only have a text-to-text connector. We should support text-to-image.

Details

The new class will live here: src/artkit/model/diffusion/gemini/

Be sure to update the docs:

Please see the tutorial: Creating New Model Classes