Allow for training loras on onle one text encoder for sdxl

This commit is contained in:
Jaret Burkett committed 2023-10-06 08:11:56 -06:00
1 parent f73402473b
commit cac8754399
5 files changed
+34 -4

No files matched your search

+1 -1
View File
@@ -60,7 +60,7 @@ def get_train_sd_device_state_preset(
preset['unet']['training'] = True
if train_lora:
preset['text_encoder']['requires_grad'] = False
# preset['text_encoder']['requires_grad'] = False
preset['unet']['requires_grad'] = False
if train_adapter: