From 35b96f32d1671eb82f98cda751c46964954e9980 Mon Sep 17 00:00:00 2001 From: kqlio67 Date: Wed, 4 Sep 2024 22:23:17 +0300 Subject: feat(Nexra): add image generation support --- g4f/models.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'g4f/models.py') diff --git a/g4f/models.py b/g4f/models.py index d0cb9f49..456a2f38 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -461,6 +461,28 @@ flux_disney = Model( ) +### ### +dalle = Model( + name = 'dalle', + base_provider = '', + best_provider = IterListProvider([Nexra]) + +) + +dalle_mini = Model( + name = 'dalle-mini', + base_provider = '', + best_provider = IterListProvider([Nexra]) + +) + +emi = Model( + name = 'emi', + base_provider = '', + best_provider = IterListProvider([Nexra]) + +) + class ModelUtils: """ Utility class for mapping string identifiers to Model instances. @@ -617,6 +639,11 @@ class ModelUtils: 'flux-3d': flux_3d, 'flux-disney': flux_disney, + +### ### +'dalle': dalle, +'dalle-mini': dalle_mini, +'emi': emi, } _all_models = list(ModelUtils.convert.keys()) -- cgit v1.2.3