sync

Author: mythz

llms/extensions/providers/chutes.py

@@ -39,6 +39,16 @@ def __init__(self, **kwargs):
                 "Animij",
                 "iLustMix",
             ]
+            imageclassic_url = "https://vonkaiser-imageclassic.chutes.ai/generate"
+            self.model_gen_urls = {
+                "flux": imageclassic_url,
+                "dreamshaper": imageclassic_url,
+                "ilustmix": imageclassic_url,
+                "juggernaut": imageclassic_url,
+                "z-image-turbo": "https://vonkaiser-z-image-turbo.chutes.ai/generate",
+                "Qwen-Image-2512": "https://vonkaiser-qwen-image-2512.chutes.ai/generate",
+                "Qwen-Image-Edit-2511": "https://vonkaiser-qwen-image-edit-2511.chutes.ai/generate"
+            }
 
         async def chat(self, chat, provider=None, context=None):
             headers = {"Authorization": f"Bearer {self.api_key}"}
@@ -53,7 +63,7 @@ async def chat(self, chat, provider=None, context=None):
             steps = self.steps
             width = self.width
             height = self.height
-            if chat["model"] == "chutes-z-image-turbo":
+            if chat["model"] == "z-image-turbo":
                 cfg_scale = min(self.cfg_scale, 5)
             payload = {
                 "model": chat["model"],
@@ -85,8 +95,8 @@ async def chat(self, chat, provider=None, context=None):
                 del payload["height"]
                 payload["size"] = aspect_ratio
 
-            gen_url = self.gen_url
-            if chat["model"].startswith("chutes-"):
+            gen_url = self.model_gen_urls.get(chat["model"])
+            if not gen_url:
                 model = payload["model"]
                 gen_url = f"https://{model}.chutes.ai/generate"
                 del payload["model"]

llms/providers-extra.json

@@ -659,7 +659,7 @@
     },
     "chutes": {
         "models": {
-            "chutes-z-image-turbo": {
+            "z-image-turbo": {
                 "name": "Z Image Turbo",
                 "modalities": {
                     "input": [
@@ -670,7 +670,7 @@
                     ]
                 }
             },
-            "qwen-image": {
+            "Qwen-Image-2512": {
                 "modalities": {
                     "input": [
                         "text"
@@ -680,8 +680,8 @@
                     ]
                 }
             },
-            "chutes-qwen-image-edit-2509": {
-                "name": "Qwen Image Edit 2509",
+            "Qwen-Image-Edit-2511": {
+                "name": "Qwen Image Edit",
                 "modalities": {
                     "input": [
                         "text",
@@ -692,8 +692,8 @@
                     ]
                 }
             },
-            "chutes-hidream": {
-                "name": "Hidream",
+            "flux": {
+                "name": "FLUX",
                 "modalities": {
                     "input": [
                         "text"
@@ -703,8 +703,8 @@
                     ]
                 }
             },
-            "chutes-hunyuan-image-3": {
-                "name": "Hunyuan Image 3",
+            "dreamshaper": {
+                "name": "Dreamshaper",
                 "modalities": {
                     "input": [
                         "text"
@@ -714,8 +714,8 @@
                     ]
                 }
             },
-            "FLUX.1-schnell": {
-                "name": "FLUX.1 Schnell",
+            "juggernaut": {
+                "name": "Juggernaut",
                 "modalities": {
                     "input": [
                         "text"
@@ -725,48 +725,8 @@
                     ]
                 }
             },
-            "chroma": {
-                "modalities": {
-                    "input": [
-                        "text"
-                    ],
-                    "output": [
-                        "image"
-                    ]
-                }
-            },
-            "JuggernautXL-Ragnarok": {
-                "name": "JuggernautXL Ragnarok",
-                "modalities": {
-                    "input": [
-                        "text"
-                    ],
-                    "output": [
-                        "image"
-                    ]
-                }
-            },
-            "Animij": {
-                "modalities": {
-                    "input": [
-                        "text"
-                    ],
-                    "output": [
-                        "image"
-                    ]
-                }
-            },
-            "Illustrij": {
-                "modalities": {
-                    "input": [
-                        "text"
-                    ],
-                    "output": [
-                        "image"
-                    ]
-                }
-            },
-            "iLustMix": {
+            "ilustmix": {
+                "name": "Ilustmix",
                 "modalities": {
                     "input": [
                         "text"