VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-16 17:02:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train_util.py

Browse Source
This commit is contained in:
DKnight54
2025-05-26 01:57:22 +08:00
committed by GitHub
parent 6f04b129df
commit c0319ea737
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/train_util.py
View File

@@ -1207,9 +1207,9 @@ class BaseDataset(torch.utils.data.Dataset):
if info.image_key in self.reg_infos:
self.reg_infos[info.image_key][0].text_encoder_outputs_npz = te_out_npz
self.reg_infos[info.image_key][0].te_cache_checked = True
self.reg_infos[info.image_key][0]. = info.text_encoder_outputs1 = hidden_state1
self.reg_infos[info.image_key][0]. = info.text_encoder_outputs2 = hidden_state2
self.reg_infos[info.image_key][0]. = info.text_encoder_pool2 = pool2
self.reg_infos[info.image_key][0].text_encoder_outputs1 = info.text_encoder_outputs1
self.reg_infos[info.image_key][0].text_encoder_outputs2 = info.text_encoder_outputs2
self.reg_infos[info.image_key][0].text_encoder_pool2 = info.text_encoder_pool2
def get_image_size(self, image_path):
return imagesize.get(image_path)