add qwen-image

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/sd_detect.py

@@ -101,6 +101,8 @@ def guess_by_name(fn, current_guess):
         return 'WanAI'
     elif 'bria' in fn.lower():
         return 'Bria'
+    elif 'qwen' in fn.lower():
+        return 'Qwen'
     return current_guess
 
 

modules/sd_models.py

@@ -368,6 +368,10 @@ def load_diffuser_force(model_type, checkpoint_info, diffusers_load_config, op='
             from pipelines.model_bria import load_bria
             sd_model = load_bria(checkpoint_info, diffusers_load_config)
             allow_post_quant = False
+        elif model_type in ['Qwen']:
+            from pipelines.model_qwen import load_qwen
+            sd_model = load_qwen(checkpoint_info, diffusers_load_config)
+            allow_post_quant = False
     except Exception as e:
         shared.log.error(f'Load {op}: path="{checkpoint_info.path}" {e}')
         if debug_load:

modules/shared_items.py

@@ -45,6 +45,7 @@ pipelines = {
     'OmniGenPipeline': getattr(diffusers, 'OmniGenPipeline', None),
     'Cosmos': getattr(diffusers, 'Cosmos2TextToImagePipeline', None),
     'WanAI': getattr(diffusers, 'WanPipeline', None),
+    'Qwen': getattr(diffusers, 'QwenImagePipeline', None),
 
     # dynamically imported and redefined later
     'Meissonic': getattr(diffusers, 'DiffusionPipeline', None),

modules/ui_extra_networks.py

@@ -932,7 +932,7 @@ def create_ui(container, button_parent, tabname, skip_indexing = False):
         return pages
 
     def ui_scan_click(title):
-        from modules.models_civitai import civit_search_metadata
+        from modules.civitai.metadata_civitai import civit_search_metadata
         civit_search_metadata(title)
         return ui_refresh_click(title)