minor updates

This commit is contained in:
Vladimir Mandic
2023-02-15 09:07:04 -05:00
parent b37ebe8b22
commit b0b326f3f8
6 changed files with 43 additions and 30 deletions
+1 -1
View File
@@ -96,7 +96,7 @@ def svd(args): # pylint: disable=redefined-outer-name
t2 = time.time()
# make state dict for LoRA
lora_network_o.apply_to(text_encoder_o, unet_o, text_encoder_different, True) # to make state dict
lora_network_o.apply_to(text_encoder_o, unet_o, text_encoder_different, True)
lora_sd = lora_network_o.state_dict()
log.info({ 'lora extracted weights': len(lora_sd), 'time': round(t2 - t1, 2) })