Commit Graph

763 Commits

Author SHA1 Message Date
fcfd2bdf8a Small cleanup. 2024-04-04 11:16:49 -04:00
0542088ef8 Refactor sampler code for more advanced sampler nodes part 2. 2024-04-04 01:26:41 -04:00
57753c964a Refactor sampling code for more advanced sampler nodes. 2024-04-03 22:09:51 -04:00
6c6a39251f Fix saving text encoder in fp8. 2024-04-02 11:46:34 -04:00
e6482fbbfc Refactor calc_cond_uncond_batch into calc_cond_batch.
calc_cond_batch can take an arbitrary amount of cond inputs.

Added a calc_cond_uncond_batch wrapper with a warning so custom nodes
won't break.
2024-04-01 18:07:47 -04:00
575acb69e4 IP2P model loading support.
This is the code to load the model and inference it with only a text
prompt. This commit does not contain the nodes to properly use it with an
image input.

This supports both the original SD1 instructpix2pix model and the
diffusers SDXL one.
2024-03-31 03:10:28 -04:00
94a5a67c32 Cleanup to support different types of inpaint models. 2024-03-29 14:44:13 -04:00
5d8898c056 Fix some performance issues with weight loading and unloading.
Lower peak memory usage when changing model.

Fix case where model weights would be unloaded and reloaded.
2024-03-28 18:04:42 -04:00
327ca1313d Support SDXS 0.9 2024-03-27 23:58:58 -04:00
ae77590b4e dora_scale support for lora file. 2024-03-25 18:09:23 -04:00
c6de09b02e Optimize memory unload strategy for more optimized performance. 2024-03-24 02:36:30 -04:00
0624838237 Add inverse noise scaling function. 2024-03-21 14:49:11 -04:00
5d875d77fe Fix regression with lcm not working with batches. 2024-03-20 20:48:54 -04:00
4b9005e949 Fix regression with model merging. 2024-03-20 13:56:12 -04:00
c18a203a8a Don't unload model weights for non weight patches. 2024-03-20 02:27:58 -04:00
150a3e946f Make LCM sampler use the model noise scaling function. 2024-03-20 01:35:59 -04:00
40e124c6be SV3D support. 2024-03-18 16:54:13 -04:00
cacb022c4a Make saved SD1 checkpoints match more closely the official one. 2024-03-18 00:26:23 -04:00
d7897fff2c Move cascade scale factor from stage_a to latent_formats.py 2024-03-16 14:49:35 -04:00
f2fe635c9f SamplerDPMAdaptative node to test the different options. 2024-03-15 22:36:10 -04:00
448d9263a2 Fix control loras breaking. 2024-03-14 09:30:21 -04:00
db8b59ecff Lower memory usage for loras in lowvram mode at the cost of perf. 2024-03-13 20:07:27 -04:00
2a813c3b09 Switch some more prints to logging. 2024-03-11 16:34:58 -04:00
0ed72befe1 Change log levels.
Logging level now defaults to info. --verbose sets it to debug.
2024-03-11 13:54:56 -04:00
65397ce601 Replace prints with logging and add --verbose argument. 2024-03-10 12:14:23 -04:00
5f60ee246e Support loading the sr cascade controlnet. 2024-03-07 01:22:48 -05:00
03e6e81629 Set upscale algorithm to bilinear for stable cascade controlnet. 2024-03-06 02:59:40 -05:00
03e83bb5d0 Support stable cascade canny controlnet. 2024-03-06 02:25:42 -05:00
10860bcd28 Add compression_ratio to controlnet code. 2024-03-05 15:15:20 -05:00
478f71a249 Remove useless check. 2024-03-04 08:51:25 -05:00
12c1080ebc Simplify differential diffusion code. 2024-03-03 15:34:42 -05:00
727021bdea Implement Differential Diffusion (#2876)
* Implement Differential Diffusion

* Cleanup.

* Fix.

* Masks should be applied at full strength.

* Fix colors.

* Register the node.

* Cleaner code.

* Fix issue with getting unipc sampler.

* Adjust thresholds.

* Switch to linear thresholds.

* Only calculate nearest_idx on valid thresholds.
2024-03-03 15:34:13 -05:00
1abf8374ec utils.set_attr can now be used to set any attribute.
The old set_attr has been renamed to set_attr_param.
2024-03-02 17:27:23 -05:00
dce3555339 Add some tesla pascal GPUs to the fp16 working but slower list. 2024-03-02 17:16:31 -05:00
51df846598 Let conditioning specify custom concat conds. 2024-03-02 11:44:06 -05:00
9f71e4b62d Let model patches patch sub objects. 2024-03-02 11:43:27 -05:00
00425563c0 Cleanup: Use sampling noise scaling function for inpainting. 2024-03-01 14:24:41 -05:00
c62e836167 Move noise scaling to object with sampling math. 2024-03-01 12:54:38 -05:00
cb7c3a2921 Allow image_only_indicator to be None. 2024-02-29 13:11:30 -05:00
b3e97fc714 Koala 700M and 1B support.
Use the UNET Loader node to load the unet file to use them.
2024-02-28 12:10:11 -05:00
37a86e4618 Remove duplicate text_projection key from some saved models. 2024-02-28 03:57:41 -05:00
8daedc5bf2 Auto detect playground v2.5 model. 2024-02-27 18:03:03 -05:00
d46583ecec Playground V2.5 support with ModelSamplingContinuousEDM node.
Use ModelSamplingContinuousEDM with edm_playground_v2.5 selected.
2024-02-27 15:12:33 -05:00
1e0fcc9a65 Make XL checkpoints save in a more standard format. 2024-02-27 02:07:40 -05:00
b416be7d78 Make the text projection saved in the checkpoint the right format. 2024-02-27 01:52:23 -05:00
03c47fc0f2 Add a min_length property to tokenizer class. 2024-02-26 21:36:37 -05:00
8ac69f62e5 Make return_projected_pooled setable from the __init__ 2024-02-25 14:49:13 -05:00
ca7c310a0e Support loading old CLIP models saved with CLIPSave. 2024-02-25 08:29:12 -05:00
c2cb8e889b Always return unprojected pooled output for gligen. 2024-02-25 07:33:13 -05:00
1cb3f6a83b Move text projection into the CLIP model code.
Fix issue with not loading the SSD1B clip correctly.
2024-02-25 01:41:08 -05:00