Skip to content

0.1.0rc2

Pre-release
Pre-release
Compare
Choose a tag to compare
@ljleb ljleb released this 01 Mar 23:12
6a46616
  • store all yaml configs directly in the repo
  • now merge methods can receive state dicts that are already loaded
  • rename merge_and_save to merge. Now the function returns a state dict by default, and will save to a file if the output keyword parameter is passed
  • move module sd_mecha.merge_methods to sd_mecha.extensions.builtin.merge_methods
  • add new merge method sd_mecha.pick_component(a, component) that rejects all keys not in the specified component. For example, pick_component(model, "diffuser") will only keep the keys in the unet of SDXL
  • rename RecipeMerger to Defaults. It is now a stand-alone utility class that simplifies merging multiple recipes in a row by specifying common values a single time
  • fix some merge methods that rely on torch.linalg and that were not playing well with threads in some environments
  • add a bunch of docstrings for public facing functions
  • function convert now can also accept a recipe node as a second parameter, removing the need to have the input dicts explicitly loaded for config auto detection or to have the configs to be specified explicitly
  • fix varargs, they were actually not working at all anymore
  • fix extensions.model_configs.serialize not handling extensions.model_configs.ModelComponent correctly
  • change default value of clip_to_ab of add_difference to False to prevent surprising behavior