VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typos

Browse Source
This commit is contained in:
Kohya S
2023-06-26 20:43:34 +09:00
parent 2c461e4ad3
commit 9ebebb22db
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_network.py
View File

@@ -98,7 +98,7 @@ class NetworkTrainer:
return False
def cache_text_encoder_outputs_if_needed(
self, args, accelerator, uner, vae, tokenizers, text_encoders, data_loader, weight_dtype
self, args, accelerator, unet, vae, tokenizers, text_encoders, data_loader, weight_dtype
):
for t_enc in text_encoders:
t_enc.to(accelerator.device)