Commit Graph

68 Commits

Author SHA1 Message Date
75b9b55b22 Fix issues with #4302 and support loading diffusers format flux. 2024-08-10 21:28:24 -04:00
1765f1c60c FLUX: Added full diffusers mapping for FLUX.1 schnell and dev. Adds full LoRA support from diffusers LoRAs. (#4302) 2024-08-10 21:26:41 -04:00
591010b7ef Support diffusers text attention flux loras. 2024-08-08 14:45:52 -04:00
78e133d041 Support simple diffusers Flux loras. 2024-08-04 22:05:48 -04:00
2ba5cc8b86 Fix some issues. 2024-08-03 15:06:40 -04:00
ba9095e5bd Automatically use fp8 for diffusion model weights if:
Checkpoint contains weights in fp8.

There isn't enough memory to load the diffusion model in GPU vram.
2024-08-03 13:45:19 -04:00
eb96c3bd82 Fix .sft file loading (they are safetensors files). 2024-08-01 11:32:58 -04:00
a3dffc447a Support AuraFlow Lora and loading model weights in diffusers format.
You can load model weights in diffusers format using the UNETLoader node.
2024-07-13 13:51:40 -04:00
f8f7568d03 Basic SD3 controlnet implementation.
Still missing the node to properly use it.
2024-06-27 18:43:11 -04:00
4ef1479dcd Multi dimension tiled scale function and tiled VAE audio encoding fallback. 2024-06-22 11:57:49 -04:00
0d6a57938e Support loading diffusers SD3 model format with UNETLoader node. 2024-06-19 22:21:18 -04:00
b08a9dd04b Remove empty line. 2024-06-19 20:20:35 -04:00
3914d5a2ae Support full SD3 loras. 2024-06-19 10:13:33 -04:00
6cd8ffc465 Reshape the empty latent image to the right amount of channels if needed. 2024-06-08 02:35:08 -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
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
b416be7d78 Make the text projection saved in the checkpoint the right format. 2024-02-27 01:52:23 -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
667c92814e Stable Cascade Stage B. 2024-02-16 13:02:03 -05:00
236bda2683 Make minimum tile size the size of the overlap. 2024-02-05 01:29:26 -05:00
9ac0b487ac Make --gpu-only put intermediate values in GPU memory instead of cpu. 2023-12-08 02:35:45 -05:00
61a123a1e0 A different way of handling multiple images passed to SVD.
Previously when a list of 3 images [0, 1, 2] was used for a 6 frame video
they were concated like this:
[0, 1, 2, 0, 1, 2]

now they are concated like this:
[0, 0, 1, 1, 2, 2]
2023-12-03 03:31:47 -05:00
c3ae99a749 Allow controlling downscale and upscale methods in PatchModelAddDownscale. 2023-11-22 03:23:16 -05:00
9f00a18095 Fix potential issues. 2023-11-16 14:59:54 -05:00
c962884a5c Make bislerp work on GPU. 2023-11-14 11:38:36 -05:00
4a8a839b40 Add option to use in place weight updating in ModelPatcher. 2023-11-11 01:11:12 -05:00
6ec3f12c6e Support SSD1B model and make it easier to support asymmetric unets. 2023-10-27 14:45:15 -04:00
d44a2de49f Make VAE code closer to sgm. 2023-10-17 15:18:51 -04:00
20d3852aa1 Pull some small changes from the other repo. 2023-10-11 20:38:48 -04:00
7c9a92f552 Don't depend on torchvision. 2023-09-19 13:12:47 -04:00
2b6b178173 Added support for lanczos scaling 2023-09-19 10:40:38 +02:00
a74c5dbf37 Move some functions to utils.py 2023-09-02 22:33:37 -04:00
77a176f9e0 Use common function to reshape batch to. 2023-09-02 03:42:49 -04:00
c77f02e1c6 Move controlnet code to comfy/controlnet.py 2023-08-25 17:33:04 -04:00
51dde87e97 Try to free enough vram for control lora inference. 2023-08-24 17:20:54 -04:00
58b2364f58 Properly support SDXL diffusers unet with UNETLoader node. 2023-07-21 14:38:56 -04:00
490771b7f4 Speed up lora loading a bit. 2023-07-15 13:25:22 -04:00
603f02d613 Fix loras not working when loading checkpoint with config. 2023-07-05 19:42:24 -04:00
af7a49916b Support loading unet files in diffusers format. 2023-07-05 17:38:59 -04:00
acf95191ff Properly support SDXL diffusers loras for unet. 2023-07-04 21:15:23 -04:00
9b93b920be Add CheckpointSave node to save checkpoints.
The created checkpoints contain workflow metadata that can be loaded by
dragging them on top of the UI or loading them with the "Load" button.

Checkpoints will be saved in fp16 or fp32 depending on the format ComfyUI
is using for inference on your hardware. To force fp32 use: --force-fp32

Anything that patches the model weights like merging or loras will be
saved.

The output directory is currently set to: output/checkpoints but that might
change in the future.
2023-06-26 12:22:27 -04:00
f87ec10a97 Support base SDXL and SDXL refiner models.
Large refactor of the model detection and loading code.
2023-06-22 13:03:50 -04:00
735ac4cf81 Remove pytorch_lightning dependency. 2023-06-13 10:11:33 -04:00
0e425603fb Small refactor. 2023-06-06 13:23:01 -04:00
a3a713b6c5 Refactor previews into one command line argument.
Clean up a few things.
2023-06-06 02:13:05 -04:00
d5a28fadaa Add latent2rgb preview 2023-06-05 18:39:56 -05:00
b4f434ee66 Preview sampled images with TAESD 2023-06-05 09:20:17 -05:00
b9818eb910 Add route to get safetensors metadata:
/view_metadata/loras?filename=lora.safetensors
2023-05-29 02:48:50 -04:00
eb4bd7711a Remove einops. 2023-05-25 18:42:56 -04:00