Merge branch 'dev' into feature/chroma-support

2026-09-20 01:31:13 +02:00 · 2025-06-26 13:06:31 -04:00
parent e91208bea9 ada38d57b4
commit d5a74d4a4a
15 changed files with 157 additions and 11 deletions
@@ -45,6 +45,8 @@ def get_model_type(pipe):
        model_type = 'sana'
    elif "HiDream" in name:
        model_type = 'h1'
+    elif "Cosmos2TextToImage" in name:
+        model_type = 'cosmos'
    # video models
    elif "CogVideo" in name:
        model_type = 'cogvideo'