prevent unintended cast for disk cached TE outputs

kohya-ss · Oct 27, 2024 · d4f7849 · d4f7849
1 parent a1255d6
commit d4f7849
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -1615,7 +1615,6 @@ def __getitem__(self, index):
                 text_encoder_outputs = self.text_encoder_output_caching_strategy.load_outputs_npz(
                     image_info.text_encoder_outputs_npz
                 )
-                text_encoder_outputs = [torch.FloatTensor(x) for x in text_encoder_outputs]
             else:
                 tokenization_required = True
             text_encoder_outputs_list.append(text_encoder_outputs)