diff options
Diffstat (limited to 'g4f/models.py')
-rw-r--r-- | g4f/models.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/g4f/models.py b/g4f/models.py index 542967f2..bfc68096 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -53,6 +53,7 @@ from .Provider import ( NexraMidjourney, NexraQwen, NexraSD15, + NexraSDLora, NexraSDTurbo, OpenaiChat, PerplexityLabs, @@ -742,10 +743,17 @@ sdxl_turbo = Model( ) +sdxl_lora = Model( + name = 'sdxl-lora', + base_provider = 'Stability AI', + best_provider = NexraSDLora + +) + sdxl = Model( name = 'sdxl', base_provider = 'Stability AI', - best_provider = IterListProvider([ReplicateHome, DeepInfraImage, sdxl_turbo.best_provider]) + best_provider = IterListProvider([ReplicateHome, DeepInfraImage, sdxl_turbo.best_provider, sdxl_lora.best_provider]) ) @@ -1111,6 +1119,7 @@ class ModelUtils: ### Stability AI ### 'sdxl': sdxl, +'sdxl-lora': sdxl_lora, 'sdxl-turbo': sdxl_turbo, 'sd-1.5': sd_1_5, 'sd-3': sd_3, |