983ebc5792
Use smart model management for VAE to decrease latency.
12 months ago
c45d1b9b67
Add a function to load a unet from a state dict.
12 months ago
f30b992b18
.sigma and .timestep now return tensors on the same device as the input.
12 months ago
13fdee6abf
Try to free memory for both cond+uncond before inference.
12 months ago
be71bb5e13
Tweak memory inference calculations a bit.
12 months ago
39e75862b2
Fix regression from last commit.
12 months ago
50dc39d6ec
Clean up the extra_options dict for the transformer patches.
...
Now everything in transformer_options gets put in extra_options.
12 months ago
5d6dfce548
Fix importing diffusers unets.
12 months ago
3e5ea74ad3
Make buggy xformers fall back on pytorch attention.
12 months ago
871cc20e13
Support SVD img2vid model.
1 year ago
410bf07771
Make VAE memory estimation take dtype into account.
1 year ago
32447f0c39
Add sampling_settings so models can specify specific sampling settings.
1 year ago
c3ae99a749
Allow controlling downscale and upscale methods in PatchModelAddDownscale.
1 year ago
72741105a6
Remove useless code.
1 year ago
6a491ebe27
Allow model config to preprocess the vae state dict on load.
1 year ago
cd4fc77d5f
Add taesd and taesdxl to VAELoader node.
...
They will show up if both the taesd_encoder and taesd_decoder or taesdxl
model files are present in the models/vae_approx directory.
1 year ago
ce67dcbcda
Make it easy for models to process the unet state dict on load.
1 year ago
d9d8702d8d
percent_to_sigma now returns a float instead of a tensor.
1 year ago
0cf4e86939
Add some command line arguments to store text encoder weights in fp8.
...
Pytorch supports two variants of fp8:
--fp8_e4m3fn-text-enc (the one that seems to give better results)
--fp8_e5m2-text-enc
1 year ago
107e78b1cb
Add support for loading SSD1B diffusers unet version.
...
Improve diffusers model detection.
1 year ago
7e3fe3ad28
Make deep shrink behave like it should.
1 year ago
9f00a18095
Fix potential issues.
1 year ago
7ea6bb038c
Print warning when controlnet can't be applied instead of crashing.
1 year ago
dcec1047e6
Invert the start and end percentages in the code.
...
This doesn't affect how percentages behave in the frontend but breaks
things if you relied on them in the backend.
percent_to_sigma goes from 0 to 1.0 instead of 1.0 to 0 for less confusion.
Make percent 0 return an extremely large sigma and percent 1.0 return a
zero one to fix imprecision.
1 year ago
57eea0efbb
heunpp2 sampler.
1 year ago
728613bb3e
Fix last pr.
1 year ago
c962884a5c
Make bislerp work on GPU.
1 year ago
420beeeb05
Clean up and refactor sampler code.
...
This should make it much easier to write custom nodes with kdiffusion type
samplers.
1 year ago
f2e49b1d57
fix: adaptation to older versions of pytroch
1 year ago
94cc718e9c
Add a way to add patches to the input block.
1 year ago
7339479b10
Disable xformers when it can't load properly.
1 year ago
4781819a85
Make memory estimation aware of model dtype.
1 year ago
dd4ba68b6e
Allow different models to estimate memory usage differently.
1 year ago
2c9dba8dc0
sampling_function now has the model object as the argument.
1 year ago
8d80584f6a
Remove useless argument from uni_pc sampler.
1 year ago
248aa3e563
Fix bug.
1 year ago
4a8a839b40
Add option to use in place weight updating in ModelPatcher.
1 year ago
412d3ff57d
Refactor.
1 year ago
58d5d71a93
Working RescaleCFG node.
...
This was broken because of recent changes so I fixed it and moved it from
the experiments repo.
1 year ago
3e0033ef30
Fix model merge bug.
...
Unload models before getting weights for model patching.
1 year ago
002aefa382
Support lcm models.
...
Use the "lcm" sampler to sample them, you also have to use the
ModelSamplingDiscrete node to set them as lcm models to use them properly.
1 year ago
ec12000136
Add support for full diff lora keys.
1 year ago
064d7583eb
Add a CONDConstant for passing non tensor conds to unet.
1 year ago
794dd2064d
Fix typo.
1 year ago
0a6fd49a3e
Print leftover keys when using the UNETLoader.
1 year ago
fe40109b57
Fix issue with object patches not being copied with patcher.
1 year ago
a527d0c795
Code refactor.
1 year ago
2a23ba0b8c
Fix unet ops not entirely on GPU.
1 year ago
844dbf97a7
Add: advanced->model->ModelSamplingDiscrete node.
...
This allows changing the sampling parameters of the model (eps or vpred)
or set the model to use zsnr.
1 year ago
656c0b5d90
CLIP code refactor and improvements.
...
More generic clip model class that can be used on more types of text
encoders.
Don't apply weighting algorithm when weight is 1.0
Don't compute an empty token output when it's not needed.
1 year ago
comfyanonymous
Jianqi Pan