Merge pull request #1021 from hitoshic1982/fix/cache-text-encoder-device-preset
Fix cache_text_encoder device preset missing in BaseModel
This commit is contained in:
1 file changed
+2
@@ -1631,6 +1631,8 @@ class BaseModel:
|
||||
active_modules = ['vae']
|
||||
if device_state_preset in ['cache_clip']:
|
||||
active_modules = ['clip']
|
||||
if device_state_preset in ['cache_text_encoder']:
|
||||
active_modules = ['text_encoder']
|
||||
if device_state_preset in ['generate']:
|
||||
active_modules = ['vae', 'unet',
|
||||
'text_encoder', 'adapter', 'refiner_unet']
|
||||
|
||||
Reference in new issue
Block a user