ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-22 15:29:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-09 13:35:04 -06:00
parent 615b0d0e94
commit 96ba2fd129
5 changed files with 205 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/stable_diffusion_model.py
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']