Upload folder using huggingface_hub

core/settings.py

@@ -17,6 +17,7 @@ _PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 _MODEL_LIST_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'model_list.yaml')
 _FILE_LIST_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'file_list.yaml')
 _CONSTANTS_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'constants.yaml')
+_MODEL_DEFAULTS_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'model_defaults.yaml')
 
 
 def load_constants_from_yaml(filepath=_CONSTANTS_PATH):
@@ -79,6 +80,13 @@ def load_models_from_yaml(model_list_filepath=_MODEL_LIST_PATH, download_map=Non
     
     return model_maps
 
+def load_model_defaults(filepath=_MODEL_DEFAULTS_PATH):
+    if not os.path.exists(filepath):
+        print(f"Warning: Model defaults file not found at {filepath}. Using empty defaults.")
+        return {}
+    with open(filepath, 'r', encoding='utf-8') as f:
+        return yaml.safe_load(f)
+
 try:
     ALL_FILE_DOWNLOAD_MAP = load_file_download_map()
     loaded_maps = load_models_from_yaml(download_map=ALL_FILE_DOWNLOAD_MAP)
@@ -86,12 +94,15 @@ try:
     ALL_MODEL_MAP = loaded_maps["ALL_MODEL_MAP"]
     
     MODEL_TYPE_MAP = {k: v[2] for k, v in ALL_MODEL_MAP.items()}
+    
+    ALL_MODEL_DEFAULTS = load_model_defaults()
 
 except Exception as e:
     print(f"FATAL: Could not load model configuration from YAML. Error: {e}")
     ALL_FILE_DOWNLOAD_MAP = {}
     MODEL_MAP_CHECKPOINT, ALL_MODEL_MAP = {}, {}
     MODEL_TYPE_MAP = {}
+    ALL_MODEL_DEFAULTS = {}
 
 
 try:

requirements.txt

@@ -1,7 +1,7 @@
-comfyui-frontend-package==1.33.10
-comfyui-workflow-templates==0.7.25
-comfyui-embedded-docs==0.3.1
-torch
+comfyui-frontend-package==1.37.11
+comfyui-workflow-templates==0.8.24
+comfyui-embedded-docs==0.4.0
+torch==2.9.1
 torchsde
 torchvision
 torchaudio
@@ -21,6 +21,8 @@ psutil
 alembic
 SQLAlchemy
 av>=14.2.0
+comfy-kitchen>=0.2.7
+requests
 
 #non essential dependencies:
 kornia>=0.7.1

ui/events.py

@@ -41,6 +41,32 @@ def load_diffsynth_controlnet_config():
         print(f"Error loading diffsynth_controlnet_models.yaml: {e}")
         return []
 
+def on_model_change(model_display_name):
+    """
+    Callback function to update UI elements when the base model changes.
+    It loads default values for steps and cfg from model_defaults.yaml.
+    """
+    defaults = ALL_MODEL_DEFAULTS.get('Default', {}).copy()
+    
+    model_found = False
+    for category, models_in_category in ALL_MODEL_DEFAULTS.items():
+        if category == 'Default' or not isinstance(models_in_category, dict):
+            continue
+            
+        if model_display_name in models_in_category:
+            if '_defaults' in models_in_category:
+                defaults.update(models_in_category['_defaults'])
+            defaults.update(models_in_category[model_display_name])
+            model_found = True
+            break
+    
+    if not model_found:
+        print(f"No specific defaults found for '{model_display_name}'. Using category or global defaults.")
+
+    steps_update = gr.update(value=defaults.get('steps'))
+    cfg_update = gr.update(value=defaults.get('cfg'))
+    
+    return steps_update, cfg_update
 
 def attach_event_handlers(ui_components, demo):
     def update_cn_input_visibility(choice):
@@ -533,6 +559,17 @@ def attach_event_handlers(ui_components, demo):
         ("txt2img", "txt2img"), ("img2img", "img2img"), ("inpaint", "inpaint"),
         ("outpaint", "outpaint"), ("hires_fix", "hires_fix"),
     ]:
+        model_dropdown = ui_components.get(f'base_model_{prefix}')
+        steps_slider = ui_components.get(f'steps_{prefix}')
+        cfg_slider = ui_components.get(f'cfg_{prefix}')
+        if all([model_dropdown, steps_slider, cfg_slider]):
+            model_dropdown.change(
+                fn=on_model_change,
+                inputs=[model_dropdown],
+                outputs=[steps_slider, cfg_slider],
+                show_progress=False
+            )
+
         if f'add_lora_button_{prefix}' in ui_components: 
             create_lora_event_handlers(prefix)
             lora_uploads = ui_components[f'lora_uploads_{prefix}']
@@ -612,7 +649,6 @@ def attach_event_handlers(ui_components, demo):
         ui_components['view_mode_inpaint'].change(fn=toggle_inpaint_fullscreen_view, inputs=[ui_components['view_mode_inpaint']], outputs=output_components, show_progress=False)
 
     def initialize_all_cn_dropdowns():
-        # Standard ControlNet
         cn_config = load_controlnet_config()
         cn_updates = {}
         if cn_config:
@@ -634,7 +670,6 @@ def attach_event_handlers(ui_components, demo):
                     for series_dd in ui_components[f'controlnet_series_{prefix}']: cn_updates[series_dd] = gr.update(choices=series_choices, value=default_series)
                     for filepath_state in ui_components[f'controlnet_filepaths_{prefix}']: cn_updates[filepath_state] = filepath
         
-        # DiffSynth ControlNet
         diffsynth_cn_config = load_diffsynth_controlnet_config()
         diffsynth_updates = {}
         if diffsynth_cn_config:

yaml/file_list.yaml

@@ -4,12 +4,16 @@ file:
       source: "hf"
       repo_id: "Comfy-Org/z_image_turbo"
       repository_file_path: "split_files/diffusion_models/z_image_turbo_bf16.safetensors"
+    - filename: "z_image_bf16.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/z_image"
+      repository_file_path: "split_files/diffusion_models/z_image_bf16.safetensors"
 
   text_encoders: 
-    - filename: "qwen_3_4b.safetensors"
+    - filename: "qwen_3_4b_fp8_mixed.safetensors"
       source: "hf"
       repo_id: "Comfy-Org/z_image_turbo"
-      repository_file_path: "split_files/text_encoders/qwen_3_4b.safetensors"
+      repository_file_path: "split_files/text_encoders/qwen_3_4b_fp8_mixed.safetensors"
 
   vae: 
     - filename: "ae.safetensors"

yaml/model_defaults.yaml

@@ -0,0 +1,17 @@
+Default:
+  steps: 25
+  cfg: 4.0
+  sampler_name: "res_multistep"
+  scheduler: "simple"
+
+Z-Image:
+  _defaults:
+    steps: 25
+    cfg: 4.0
+    sampler_name: "res_multistep"
+    scheduler: "simple"
+  "Tongyi-MAI/Z Image Turbo":
+    steps: 9
+    cfg: 1.0
+    sampler_name: "res_multistep"
+    scheduler: "simple"

yaml/model_list.yaml

@@ -3,4 +3,9 @@ Checkpoint:
     components:
       unet: "z_image_turbo_bf16.safetensors"
       vae: "ae.safetensors"
-      clip: "qwen_3_4b.safetensors"
+      clip: "qwen_3_4b_fp8_mixed.safetensors"
+  - display_name: "Tongyi-MAI/Z Image"
+    components:
+      unet: "z_image_bf16.safetensors"
+      vae: "ae.safetensors"
+      clip: "qwen_3_4b_fp8_mixed.safetensors"