diff options
Diffstat (limited to 'training')
-rw-r--r-- | training/util.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/training/util.py b/training/util.py index e8d22ae..000173d 100644 --- a/training/util.py +++ b/training/util.py | |||
@@ -44,7 +44,6 @@ class CheckpointerBase: | |||
44 | self, | 44 | self, |
45 | datamodule, | 45 | datamodule, |
46 | output_dir: Path, | 46 | output_dir: Path, |
47 | instance_identifier, | ||
48 | placeholder_token, | 47 | placeholder_token, |
49 | placeholder_token_id, | 48 | placeholder_token_id, |
50 | sample_image_size, | 49 | sample_image_size, |
@@ -54,7 +53,6 @@ class CheckpointerBase: | |||
54 | ): | 53 | ): |
55 | self.datamodule = datamodule | 54 | self.datamodule = datamodule |
56 | self.output_dir = output_dir | 55 | self.output_dir = output_dir |
57 | self.instance_identifier = instance_identifier | ||
58 | self.placeholder_token = placeholder_token | 56 | self.placeholder_token = placeholder_token |
59 | self.placeholder_token_id = placeholder_token_id | 57 | self.placeholder_token_id = placeholder_token_id |
60 | self.sample_image_size = sample_image_size | 58 | self.sample_image_size = sample_image_size |
@@ -90,7 +88,7 @@ class CheckpointerBase: | |||
90 | if j * data.batch_size < self.sample_batch_size * self.sample_batches | 88 | if j * data.batch_size < self.sample_batch_size * self.sample_batches |
91 | ] | 89 | ] |
92 | prompts = [ | 90 | prompts = [ |
93 | prompt.format(identifier=self.instance_identifier) | 91 | prompt |
94 | for batch in batches | 92 | for batch in batches |
95 | for prompt in batch["prompts"] | 93 | for prompt in batch["prompts"] |
96 | ] | 94 | ] |