Skip to content

Commit

Permalink
Move model merging nodes to advanced and add to readme.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Jun 30, 2023
1 parent 5a9ddf9 commit 9f29863
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ This ui will let you design and execute advanced stable diffusion pipelines usin
- [Upscale Models (ESRGAN, ESRGAN variants, SwinIR, Swin2SR, etc...)](https://comfyanonymous.github.io/ComfyUI_examples/upscale_models/)
- [unCLIP Models](https://comfyanonymous.github.io/ComfyUI_examples/unclip/)
- [GLIGEN](https://comfyanonymous.github.io/ComfyUI_examples/gligen/)
- Latent previews with [TAESD](https://github.com/madebyollin/taesd)
- [Model Merging](https://comfyanonymous.github.io/ComfyUI_examples/model_merging/)
- Latent previews with [TAESD](#how-to-show-high-quality-previews)
- Starts up very fast.
- Works fully offline: will never download anything.
- [Config file](extra_model_paths.yaml.example) to set the search paths for models.
Expand Down
6 changes: 3 additions & 3 deletions comfy_extras/nodes_model_merging.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ def INPUT_TYPES(s):
RETURN_TYPES = ("MODEL",)
FUNCTION = "merge"

CATEGORY = "_for_testing/model_merging"
CATEGORY = "advanced/model_merging"

def merge(self, model1, model2, ratio):
m = model1.clone()
Expand All @@ -35,7 +35,7 @@ def INPUT_TYPES(s):
RETURN_TYPES = ("MODEL",)
FUNCTION = "merge"

CATEGORY = "_for_testing/model_merging"
CATEGORY = "advanced/model_merging"

def merge(self, model1, model2, **kwargs):
m = model1.clone()
Expand Down Expand Up @@ -68,7 +68,7 @@ def INPUT_TYPES(s):
FUNCTION = "save"
OUTPUT_NODE = True

CATEGORY = "_for_testing/model_merging"
CATEGORY = "advanced/model_merging"

def save(self, model, clip, vae, filename_prefix, prompt=None, extra_pnginfo=None):
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
Expand Down

0 comments on commit 9f29863

Please sign in to comment.