pythongosssss 7bf381bc9e Add model management and database ...

- use sqlalchemy + alembic + sqlite for db
- extract model data and previews
- endpoints for db interactions
- add tests

2025-03-28 11:39:56 +08:00

..

cldm

Replace print with logging (#6138)

2024-12-20 16:24:55 -05:00

comfy_types

Fix custom node type-hinting examples (#6281)

2024-12-31 03:41:09 -05:00

extra_samplers

Fix and enforce no trailing whitespace.

2024-12-31 03:16:37 -05:00

k_diffusion

Rewrite res_multistep sampler and implement res_multistep_cfg_pp sampler. (#6462)

2025-01-14 18:20:06 -05:00

ldm

Remove useless code.

2025-01-13 03:52:37 -05:00

sd1_tokenizer

Initial commit.

2023-01-16 22:37:14 -05:00

t2i_adapter

Controlnet refactor.

2024-06-27 18:43:11 -04:00

taesd

Improvements to the TAESD3 implementation.

2024-06-16 02:04:24 -04:00

text_encoders

WIP support for Nvidia Cosmos 7B and 14B text to world (video) models.

2025-01-10 09:14:16 -05:00

checkpoint_pickle.py

Remove pytorch_lightning dependency.

2023-06-13 10:11:33 -04:00

cli_args.py

Add model management and database

2025-03-28 11:39:56 +08:00

clip_config_bigg.json

Fix potential issue with non clip text embeddings.

2024-07-30 14:41:13 -04:00

clip_model.py

Support new flux model variants.

2024-11-21 08:38:23 -05:00

clip_vision_config_g.json

Add support for clip g vision model to CLIPVisionLoader.

2023-08-18 11:13:29 -04:00

clip_vision_config_h.json

Add support for unCLIP SD2.x models.

2023-04-01 23:19:15 -04:00

clip_vision_config_vitl_336.json

support clip-vit-large-patch14-336 (#4042)

2024-07-17 13:12:50 -04:00

clip_vision_config_vitl.json

Add support for unCLIP SD2.x models.

2023-04-01 23:19:15 -04:00

clip_vision_siglip_384.json

Support new flux model variants.

2024-11-21 08:38:23 -05:00

clip_vision.py

Add a way to disable cropping in the CLIPVisionEncode node.

2024-11-28 20:24:47 -05:00

conds.py

Fix bug with cosxl edit model.

2024-06-10 16:00:03 -04:00

controlnet.py

Remove some trailing white space.

2024-12-27 18:02:30 -05:00

diffusers_convert.py

Basic Hunyuan Video model support.

2024-12-16 19:35:40 -05:00

diffusers_load.py

load_unet -> load_diffusion_model with a model_options argument.

2024-08-12 23:20:57 -04:00

float.py

Clamp output when rounding weight to prevent Nan.

2024-10-19 19:07:10 -04:00

gligen.py

Lint and fix undefined names (1/N) (#6028)

2024-12-12 18:55:26 -05:00

hooks.py

Hooks Part 2 - TransformerOptionsHook and AdditionalModelsHook (#6377)

2025-01-11 12:20:23 -05:00

latent_formats.py

Fast latent preview for Cosmos.

2025-01-11 04:41:24 -05:00

lora_convert.py

Support new flux model variants.

2024-11-21 08:38:23 -05:00

lora.py

Add PixArt model support (#6055)

2024-12-20 15:25:00 -05:00

model_base.py

Implement Cosmos Image/Video to World (Video) diffusion models.

2025-01-14 05:14:10 -05:00

model_detection.py

Implement Cosmos Image/Video to World (Video) diffusion models.

2025-01-14 05:14:10 -05:00

model_management.py

Remove old unused function.

2025-01-04 07:20:54 -05:00

model_patcher.py

Hooks Part 2 - TransformerOptionsHook and AdditionalModelsHook (#6377)

2025-01-11 12:20:23 -05:00

model_sampling.py

Make timestep ranges more usable on rectified flow models.

2024-12-05 16:40:58 -05:00

ops.py

Make fast fp8 take a bit less peak memory.

2024-12-24 18:05:19 -05:00

options.py

Only parse command line args when main.py is called.

2023-09-13 11:38:20 -04:00

patcher_extension.py

Add ruff rule for empty line with trailing whitespace.

2024-12-28 05:23:08 -05:00

sample.py

Auto reshape 2d to 3d latent for single image generation on video model.

2024-12-29 02:26:49 -05:00

sampler_helpers.py

Hooks Part 2 - TransformerOptionsHook and AdditionalModelsHook (#6377)

2025-01-11 12:20:23 -05:00

samplers.py

Rewrite res_multistep sampler and implement res_multistep_cfg_pp sampler. (#6462)

2025-01-14 18:20:06 -05:00

sd1_clip_config.json

Fix potential issue with non clip text embeddings.

2024-07-30 14:41:13 -04:00

sd1_clip.py

Remove unsafe embedding load for very old pytorch.

2025-01-15 04:32:23 -05:00

sd.py

Implement Cosmos Image/Video to World (Video) diffusion models.

2025-01-14 05:14:10 -05:00

sdxl_clip.py

Long CLIP L support for SDXL, SD3 and Flux.

2024-09-15 07:59:38 -04:00

supported_models_base.py

Mixed precision diffusion models with scaled fp8.

2024-10-21 18:12:51 -04:00

supported_models.py

Implement Cosmos Image/Video to World (Video) diffusion models.

2025-01-14 05:14:10 -05:00

utils.py

Force safe loading of files in torch format on pytorch 2.4+

2025-01-15 03:50:27 -05:00