Daniel Gu e62b32aadc Add self.image_encoder, self.text_decoder to list of models to offload to CPU in the enable_sequential_cpu_offload(...)/enable_model_cpu_offload(...) methods to make test_cpu_offload_forward_pass pass.

2023-05-05 12:12:42 -07:00

..

commands

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

experimental

Ruff: apply same rules as in transformers (#2827)

2023-03-27 16:18:57 +02:00

models

Add upsample_size to AttnUpBlock2D, AttnDownBlock2D (#3275)

2023-05-05 19:50:41 +01:00

pipelines

Add self.image_encoder, self.text_decoder to list of models to offload to CPU in the enable_sequential_cpu_offload(...)/enable_model_cpu_offload(...) methods to make test_cpu_offload_forward_pass pass.

2023-05-05 12:12:42 -07:00

schedulers

Add use_Karras_sigmas to LMSDiscreteScheduler (#3351)

2023-05-06 12:19:27 +01:00

utils

Torch compile graph fix (#3286)

2023-05-05 07:23:51 -07:00

__init__.py

Fix code with make style.

2023-05-05 07:55:15 -07:00

configuration_utils.py

add model (#3230)

2023-05-05 07:23:50 -07:00

dependency_versions_check.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

dependency_versions_table.py

[Quality] Make style (#3341)

2023-05-05 07:23:51 -07:00

image_processor.py

Postprocessing refactor img2img (#3268)

2023-05-05 07:23:51 -07:00

loaders.py

[LoRA] quality of life improvements in the loading semantics and docs (#3180)

2023-05-05 07:23:51 -07:00

optimization.py

add constant learning rate with custom rule (#3133)

2023-05-05 07:23:51 -07:00

pipeline_utils.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

training_utils.py

ema step, don't empty cuda cache (#2563)

2023-03-06 10:54:56 +01:00