Added methods to the dataloader to automatically generate controls for line, mask, inpainting, depth, and pose.

This commit is contained in:
Jaret Burkett committed 2025-04-09 13:35:04 -06:00
1 parent 615b0d0e94
commit 96ba2fd129
5 files changed
+205 -11

No files matched your search

+2
View File
@@ -3009,6 +3009,8 @@ class StableDiffusion:
active_modules = ['vae']
if device_state_preset in ['cache_clip']:
active_modules = ['clip']
if device_state_preset in ['unload']:
active_modules = []
if device_state_preset in ['generate']:
active_modules = ['vae', 'unet', 'text_encoder', 'adapter', 'refiner_unet']