Skip to content

Commit 662172b

Browse files
authored
Merge pull request #17 from ai-forever/dev
imagename_column overwrite fix
2 parents 76dcc32 + 4435c3e commit 662172b

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

DPF/processors/text2image/t2i_processor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -288,6 +288,8 @@ def to_shards(
288288
) -> None:
289289
if columns_to_save is None:
290290
columns_to_save = []
291+
elif self.imagename_column in columns_to_save:
292+
columns_to_save.remove(self.imagename_column)
291293
dataloader = UniversalT2IDataloader(
292294
self.filesystem,
293295
self.df,

0 commit comments

Comments
 (0)