diff --git a/examples/README.md b/examples/README.md index 5945c8c..426a25a 100644 --- a/examples/README.md +++ b/examples/README.md @@ -75,11 +75,11 @@ Times are approximate on an RTX 5060 Ti 16 GB, not counting the first model load |---|---| | `moss_image_sft_int8_convrot.safetensors` | `models/diffusion_models/` | | `qwen3.5_9b_w4a8_gptq.safetensors` | `models/text_encoders/` | -| `flux2-vae_small.safetensors` | `models/vae/` | +| `flux2-vae.safetensors` | `models/vae/` | * **What the grid tests:** two config arrays with the same prompt and settings (er_sde / simple, 25 steps, cfg 2.5): one with Skip Layer Guidance on, one off. **2 images** * **Time:** TBD -* **Notes:** SLG uses the official recipe, single block 22 for the first 0-50 % of the steps. Compare the two images to see what it does. CLIP type `moss_image`. The 9B text encoder is large; on a 16 GB card expect model swapping. +* **Notes:** SLG uses the official recipe, single block 22 for the first 0-50 % of the steps. Compare the two images to see what it does. CLIP type `moss_image`. A smaller `flux2-vae_small.safetensors` VAE also exists and can be picked in the Builder instead. The 9B text encoder is large; on a 16 GB card expect model swapping. ## Make your own diff --git a/examples/uscg-moss-image.json b/examples/uscg-moss-image.json index 852abfa..58bc0d9 100644 --- a/examples/uscg-moss-image.json +++ b/examples/uscg-moss-image.json @@ -123,7 +123,7 @@ "simple", "25", "2.5", - "{\n \"session_name\": \"example_moss_image\",\n \"config_name\": \"example_moss_image\",\n \"auto_save\": false,\n \"include_none\": false,\n \"label_mode\": false,\n \"global_positive_groups\": [\n [\n \"a cozy reading nook by a rain-streaked window, warm lamp light, soft focus, detailed\"\n ]\n ],\n \"global_negative\": \"\",\n \"distribution_enabled\": false,\n \"worker_urls\": [],\n \"claim_timeout\": 600,\n \"use_master_encoding\": false,\n \"config_arrays\": [\n {\n \"name\": \"MOSS-Image, SLG on (block 22, 0-50%)\",\n \"samplers\": [\n \"er_sde\"\n ],\n \"schedulers\": [\n \"simple\"\n ],\n \"steps\": \"25\",\n \"cfg\": \"2.5\",\n \"models\": [\n {\n \"path\": \"moss_image_sft_int8_convrot.safetensors\",\n \"type\": \"diffusion_model\"\n }\n ],\n \"vaes\": [\n \"flux2-vae_small.safetensors\"\n ],\n \"text_encoders\": [\n \"qwen3.5_9b_w4a8_gptq.safetensors\"\n ],\n \"clip_type\": \"moss_image\",\n \"gguf_options\": {},\n \"loras\": [\n \"None\"\n ],\n \"lora_omit_triggers\": [],\n \"lora_triggerwords_append_settings\": {},\n \"lora_bypass_states\": {},\n \"lora_strength_lock\": {},\n \"model_bypass_states\": {},\n \"vae_bypass_states\": {},\n \"te_bypass_states\": {},\n \"combine\": false,\n \"positive_prompt_groups\": [],\n \"negative_prompt\": \"\",\n \"use_custom_prompts\": false,\n \"model_prompt_prefix\": \"\",\n \"model_prompt_suffix\": \"\",\n \"attention_modes\": [\n \"default\"\n ],\n \"model_sampling_override\": \"none\",\n \"model_sampling_shift\": \"1.73\",\n \"model_sampling_flux_max_shift\": \"1.15\",\n \"model_sampling_flux_base_shift\": \"0.5\",\n \"use_advanced_sampling\": false,\n \"advanced_guider\": \"cfg_guider\",\n \"advanced_scheduler\": \"basic\",\n \"use_flux_guidance\": false,\n \"flux_guidance_value\": \"3.5\",\n \"use_deep_shrink\": false,\n \"deep_shrink_block_number\": 3,\n \"deep_shrink_downscale_factor\": 2.0,\n \"deep_shrink_start_percent\": 0.0,\n \"deep_shrink_end_percent\": 0.35,\n \"deep_shrink_downscale_after_skip\": true,\n \"deep_shrink_downscale_method\": \"bicubic\",\n \"deep_shrink_upscale_method\": \"bicubic\",\n \"use_slg\": true,\n \"slg_double_layers\": \"\",\n \"slg_single_layers\": \"22\",\n \"slg_start_percent\": 0.0,\n \"slg_end_percent\": 0.5,\n \"seed_behavior\": \"fixed\",\n \"resolutions\": [\n \"1024x1024\"\n ]\n },\n {\n \"name\": \"MOSS-Image, SLG off\",\n \"samplers\": [\n \"er_sde\"\n ],\n \"schedulers\": [\n \"simple\"\n ],\n \"steps\": \"25\",\n \"cfg\": \"2.5\",\n \"models\": [\n {\n \"path\": \"moss_image_sft_int8_convrot.safetensors\",\n \"type\": \"diffusion_model\"\n }\n ],\n \"vaes\": [\n \"flux2-vae_small.safetensors\"\n ],\n \"text_encoders\": [\n \"qwen3.5_9b_w4a8_gptq.safetensors\"\n ],\n \"clip_type\": \"moss_image\",\n \"gguf_options\": {},\n \"loras\": [\n \"None\"\n ],\n \"lora_omit_triggers\": [],\n \"lora_triggerwords_append_settings\": {},\n \"lora_bypass_states\": {},\n \"lora_strength_lock\": {},\n \"model_bypass_states\": {},\n \"vae_bypass_states\": {},\n \"te_bypass_states\": {},\n \"combine\": false,\n \"positive_prompt_groups\": [],\n \"negative_prompt\": \"\",\n \"use_custom_prompts\": false,\n \"model_prompt_prefix\": \"\",\n \"model_prompt_suffix\": \"\",\n \"attention_modes\": [\n \"default\"\n ],\n \"model_sampling_override\": \"none\",\n \"model_sampling_shift\": \"1.73\",\n \"model_sampling_flux_max_shift\": \"1.15\",\n \"model_sampling_flux_base_shift\": \"0.5\",\n \"use_advanced_sampling\": false,\n \"advanced_guider\": \"cfg_guider\",\n \"advanced_scheduler\": \"basic\",\n \"use_flux_guidance\": false,\n \"flux_guidance_value\": \"3.5\",\n \"use_deep_shrink\": false,\n \"deep_shrink_block_number\": 3,\n \"deep_shrink_downscale_factor\": 2.0,\n \"deep_shrink_start_percent\": 0.0,\n \"deep_shrink_end_percent\": 0.35,\n \"deep_shrink_downscale_after_skip\": true,\n \"deep_shrink_downscale_method\": \"bicubic\",\n \"deep_shrink_upscale_method\": \"bicubic\",\n \"use_slg\": false,\n \"slg_double_layers\": \"\",\n \"slg_single_layers\": \"22\",\n \"slg_start_percent\": 0.0,\n \"slg_end_percent\": 0.5,\n \"seed_behavior\": \"fixed\",\n \"resolutions\": [\n \"1024x1024\"\n ]\n }\n ],\n \"upscaling\": {\n \"enabled\": false,\n \"save_pre_upscale\": false,\n \"run_upscales_at_end\": false,\n \"hires_prompt_adjust\": false,\n \"hires_prompt_behavior\": \"append_end\",\n \"hires_prompt_text\": \"\",\n \"pipelines\": [\n {\n \"active\": true,\n \"name\": \"Pipeline 1\",\n \"steps\": [\n {\n \"active\": true,\n \"mode\": \"hires_only\",\n \"repeat\": 1,\n \"upscale_models\": [],\n \"upscale_ratios\": \"1.5\",\n \"upscale_size\": \"2.0\",\n \"hires_denoise\": \"0.3\",\n \"hires_steps\": 0,\n \"tiled_vae\": false,\n \"tile_size\": 512,\n \"tile_overlap\": 64,\n \"temporal_size\": 512,\n \"temporal_overlap\": 64,\n \"resize_method\": \"bilinear\",\n \"hires_tiled_sampling\": false,\n \"hires_tile_width\": 512,\n \"hires_tile_height\": 512,\n \"hires_mask_blur\": 8,\n \"hires_tile_padding\": 32,\n \"hires_force_uniform_tiles\": false\n }\n ]\n }\n ]\n },\n \"cooldown\": {\n \"enabled\": false,\n \"seconds\": 5,\n \"every_n\": 1,\n \"clear_vram\": false\n },\n \"image_format\": \"webp\",\n \"overwrite_existing\": false,\n \"flush_batch_every\": 1,\n \"lora_triggerwords_mode\": \"None\",\n \"save_conditioning_cache_to_file\": false,\n \"enable_model_cache\": false,\n \"vae_batch_size\": 1,\n \"add_random_seeds_to_gens\": 0\n}", + "{\n \"session_name\": \"example_moss_image\",\n \"config_name\": \"example_moss_image\",\n \"auto_save\": false,\n \"include_none\": false,\n \"label_mode\": false,\n \"global_positive_groups\": [\n [\n \"a cozy reading nook by a rain-streaked window, warm lamp light, soft focus, detailed\"\n ]\n ],\n \"global_negative\": \"\",\n \"distribution_enabled\": false,\n \"worker_urls\": [],\n \"claim_timeout\": 600,\n \"use_master_encoding\": false,\n \"config_arrays\": [\n {\n \"name\": \"MOSS-Image, SLG on (block 22, 0-50%)\",\n \"samplers\": [\n \"er_sde\"\n ],\n \"schedulers\": [\n \"simple\"\n ],\n \"steps\": \"25\",\n \"cfg\": \"2.5\",\n \"models\": [\n {\n \"path\": \"moss_image_sft_int8_convrot.safetensors\",\n \"type\": \"diffusion_model\"\n }\n ],\n \"vaes\": [\n \"flux2-vae.safetensors\"\n ],\n \"text_encoders\": [\n \"qwen3.5_9b_w4a8_gptq.safetensors\"\n ],\n \"clip_type\": \"moss_image\",\n \"gguf_options\": {},\n \"loras\": [\n \"None\"\n ],\n \"lora_omit_triggers\": [],\n \"lora_triggerwords_append_settings\": {},\n \"lora_bypass_states\": {},\n \"lora_strength_lock\": {},\n \"model_bypass_states\": {},\n \"vae_bypass_states\": {},\n \"te_bypass_states\": {},\n \"combine\": false,\n \"positive_prompt_groups\": [],\n \"negative_prompt\": \"\",\n \"use_custom_prompts\": false,\n \"model_prompt_prefix\": \"\",\n \"model_prompt_suffix\": \"\",\n \"attention_modes\": [\n \"default\"\n ],\n \"model_sampling_override\": \"none\",\n \"model_sampling_shift\": \"1.73\",\n \"model_sampling_flux_max_shift\": \"1.15\",\n \"model_sampling_flux_base_shift\": \"0.5\",\n \"use_advanced_sampling\": false,\n \"advanced_guider\": \"cfg_guider\",\n \"advanced_scheduler\": \"basic\",\n \"use_flux_guidance\": false,\n \"flux_guidance_value\": \"3.5\",\n \"use_deep_shrink\": false,\n \"deep_shrink_block_number\": 3,\n \"deep_shrink_downscale_factor\": 2.0,\n \"deep_shrink_start_percent\": 0.0,\n \"deep_shrink_end_percent\": 0.35,\n \"deep_shrink_downscale_after_skip\": true,\n \"deep_shrink_downscale_method\": \"bicubic\",\n \"deep_shrink_upscale_method\": \"bicubic\",\n \"use_slg\": true,\n \"slg_double_layers\": \"\",\n \"slg_single_layers\": \"22\",\n \"slg_start_percent\": 0.0,\n \"slg_end_percent\": 0.5,\n \"seed_behavior\": \"fixed\",\n \"resolutions\": [\n \"1024x1024\"\n ]\n },\n {\n \"name\": \"MOSS-Image, SLG off\",\n \"samplers\": [\n \"er_sde\"\n ],\n \"schedulers\": [\n \"simple\"\n ],\n \"steps\": \"25\",\n \"cfg\": \"2.5\",\n \"models\": [\n {\n \"path\": \"moss_image_sft_int8_convrot.safetensors\",\n \"type\": \"diffusion_model\"\n }\n ],\n \"vaes\": [\n \"flux2-vae.safetensors\"\n ],\n \"text_encoders\": [\n \"qwen3.5_9b_w4a8_gptq.safetensors\"\n ],\n \"clip_type\": \"moss_image\",\n \"gguf_options\": {},\n \"loras\": [\n \"None\"\n ],\n \"lora_omit_triggers\": [],\n \"lora_triggerwords_append_settings\": {},\n \"lora_bypass_states\": {},\n \"lora_strength_lock\": {},\n \"model_bypass_states\": {},\n \"vae_bypass_states\": {},\n \"te_bypass_states\": {},\n \"combine\": false,\n \"positive_prompt_groups\": [],\n \"negative_prompt\": \"\",\n \"use_custom_prompts\": false,\n \"model_prompt_prefix\": \"\",\n \"model_prompt_suffix\": \"\",\n \"attention_modes\": [\n \"default\"\n ],\n \"model_sampling_override\": \"none\",\n \"model_sampling_shift\": \"1.73\",\n \"model_sampling_flux_max_shift\": \"1.15\",\n \"model_sampling_flux_base_shift\": \"0.5\",\n \"use_advanced_sampling\": false,\n \"advanced_guider\": \"cfg_guider\",\n \"advanced_scheduler\": \"basic\",\n \"use_flux_guidance\": false,\n \"flux_guidance_value\": \"3.5\",\n \"use_deep_shrink\": false,\n \"deep_shrink_block_number\": 3,\n \"deep_shrink_downscale_factor\": 2.0,\n \"deep_shrink_start_percent\": 0.0,\n \"deep_shrink_end_percent\": 0.35,\n \"deep_shrink_downscale_after_skip\": true,\n \"deep_shrink_downscale_method\": \"bicubic\",\n \"deep_shrink_upscale_method\": \"bicubic\",\n \"use_slg\": false,\n \"slg_double_layers\": \"\",\n \"slg_single_layers\": \"22\",\n \"slg_start_percent\": 0.0,\n \"slg_end_percent\": 0.5,\n \"seed_behavior\": \"fixed\",\n \"resolutions\": [\n \"1024x1024\"\n ]\n }\n ],\n \"upscaling\": {\n \"enabled\": false,\n \"save_pre_upscale\": false,\n \"run_upscales_at_end\": false,\n \"hires_prompt_adjust\": false,\n \"hires_prompt_behavior\": \"append_end\",\n \"hires_prompt_text\": \"\",\n \"pipelines\": [\n {\n \"active\": true,\n \"name\": \"Pipeline 1\",\n \"steps\": [\n {\n \"active\": true,\n \"mode\": \"hires_only\",\n \"repeat\": 1,\n \"upscale_models\": [],\n \"upscale_ratios\": \"1.5\",\n \"upscale_size\": \"2.0\",\n \"hires_denoise\": \"0.3\",\n \"hires_steps\": 0,\n \"tiled_vae\": false,\n \"tile_size\": 512,\n \"tile_overlap\": 64,\n \"temporal_size\": 512,\n \"temporal_overlap\": 64,\n \"resize_method\": \"bilinear\",\n \"hires_tiled_sampling\": false,\n \"hires_tile_width\": 512,\n \"hires_tile_height\": 512,\n \"hires_mask_blur\": 8,\n \"hires_tile_padding\": 32,\n \"hires_force_uniform_tiles\": false\n }\n ]\n }\n ]\n },\n \"cooldown\": {\n \"enabled\": false,\n \"seconds\": 5,\n \"every_n\": 1,\n \"clear_vram\": false\n },\n \"image_format\": \"webp\",\n \"overwrite_existing\": false,\n \"flush_batch_every\": 1,\n \"lora_triggerwords_mode\": \"None\",\n \"save_conditioning_cache_to_file\": false,\n \"enable_model_cache\": false,\n \"vae_batch_size\": 1,\n \"add_random_seeds_to_gens\": 0\n}", false, "" ] @@ -275,7 +275,7 @@ "outputs": [], "properties": {}, "widgets_values": [ - "### MOSS-Image example\nNeeds: `flux2-vae_small.safetensors`, `moss_image_sft_int8_convrot.safetensors`, `qwen3.5_9b_w4a8_gptq.safetensors`\n\nSee `examples/README.md` for the folder each file goes in. Missing files show up as missing in the Config Builder; pick your own equivalents there." + "### MOSS-Image example\nNeeds: `flux2-vae.safetensors`, `moss_image_sft_int8_convrot.safetensors`, `qwen3.5_9b_w4a8_gptq.safetensors`\n\nSee `examples/README.md` for the folder each file goes in. Missing files show up as missing in the Config Builder; pick your own equivalents there." ] } ], diff --git a/tests/test_examples.py b/tests/test_examples.py index 72fdef8..9186b81 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -48,7 +48,7 @@ ALLOWED_FILES = { "uscg-z-image-turbo.json": {"z_image_turbo_bf16.safetensors", "qwen_3_4b.safetensors", "ae.safetensors"}, "uscg-krea2-turbo.json": {"krea2_turbo_fp8_scaled.safetensors", "qwen3vl_4b_fp8_scaled.safetensors", "qwen_image_vae.safetensors"}, "uscg-flux2-klein-9b.json": {"flux-2-klein-9b-fp8.safetensors", "qwen_3_8b_fp8mixed.safetensors", "flux2-vae.safetensors"}, - "uscg-moss-image.json": {"moss_image_sft_int8_convrot.safetensors", "qwen3.5_9b_w4a8_gptq.safetensors", "flux2-vae_small.safetensors"}, + "uscg-moss-image.json": {"moss_image_sft_int8_convrot.safetensors", "qwen3.5_9b_w4a8_gptq.safetensors", "flux2-vae.safetensors"}, }