Commit Graph

24 Commits (ac2da7e9bf8a0ffcd3dbcab738e2b3e631454ffd)

Author SHA1 Message Date
Uminosachi 79bb14139b Fix linting warnings and errors using Ruff 2024-05-08 14:22:17 +09:00
Uminosachi f0d333ba36 Update decorator to auto-identify generator func 2023-09-12 10:08:33 +09:00
Uminosachi 7a0c424303 Optimize to reduce unnecessary CPU offloading 2023-09-11 19:16:35 +09:00
Uminosachi a868cf8b5f Offload when lowvram and medvram are not added 2023-09-11 13:51:37 +09:00
Uminosachi 4650c8aed3 Fix not to offload if shared.sd_model is SDXL 2023-09-11 12:52:13 +09:00
Uminosachi 4794183f4e Add decorator that support yield from nest 2023-08-28 17:12:28 +09:00
Uminosachi 78f6832032 Use ContextDecorator for switching torch.load 2023-08-24 09:18:32 +09:00
Uminosachi a7d836c2aa Remove patch as PR has been merged to web UI dev 2023-08-21 19:18:30 +09:00
Uminosachi 5df11f37eb Fix mismatch between model and VAE post-backup 2023-08-19 21:24:07 +09:00
Uminosachi abf3e248a1 Add offload reload decorator and refactor 2023-08-15 09:17:30 +09:00
Uminosachi f98e735cb9 Remove patch due to PR merged to web UI dev 2023-08-10 09:07:45 +09:00
Uminosachi 9154b52c96 Revert sd_model original device after CPU offload 2023-08-09 09:20:19 +09:00
Uminosachi 2f0428721c Fix mismatch between sd_model opts in web UI dev 2023-08-08 11:58:28 +09:00
Uminosachi 747899520b Remove function for unloading model weights 2023-08-08 06:50:52 +09:00
Uminosachi 8888d1973e Support model reuse in web UI dev branch 2023-08-06 18:53:17 +09:00
Uminosachi f30a978d7f Fix memory usage growth in web UI dev branch 2023-08-06 11:14:42 +09:00
Uminosachi 29a9be294c Add code to backup and load checkpoint info 2023-08-05 20:08:42 +09:00
Uminosachi 68c50a88ed Avoid reload model func call if sd_model is None 2023-08-05 15:48:07 +09:00
Uminosachi 0940bbba4d Add buttons to get txt2img and img2img prompts 2023-08-03 14:48:48 +09:00
Uminosachi 04e8e8b507 Fix model loading while clearing torch cache 2023-08-02 20:11:52 +09:00
Uminosachi a971a38a51 Fix linting errors and warnings using Flake8 2023-07-17 14:08:42 +09:00
Uminosachi 52268b8075 Disable SAM button until image upload is complete 2023-07-01 15:26:20 +09:00
Uminosachi 8d261b4293 Add cache clearing decorator to event function 2023-07-01 13:18:03 +09:00
Uminosachi 5c2fbf2db9 Move threading of model load to another file 2023-07-01 08:09:54 +09:00