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

prevent unintended cast for disk cached TE outputs

Browse Source
This commit is contained in:
kohya-ss
2024-10-27 19:35:56 +09:00
parent a1255d637f
commit d4f7849592
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
library/train_util.py
View File

@@ -1615,7 +1615,6 @@ class BaseDataset(torch.utils.data.Dataset):
text_encoder_outputs = self.text_encoder_output_caching_strategy.load_outputs_npz( text_encoder_outputs = self.text_encoder_output_caching_strategy.load_outputs_npz(
image_info.text_encoder_outputs_npz image_info.text_encoder_outputs_npz
) )
text_encoder_outputs = [torch.FloatTensor(x) for x in text_encoder_outputs]
else: else:
tokenization_required = True tokenization_required = True
text_encoder_outputs_list.append(text_encoder_outputs) text_encoder_outputs_list.append(text_encoder_outputs)