Improved class prompt handling

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:
         self,
         datamodule,
         output_dir: Path,
-        instance_identifier,
         placeholder_token,
         placeholder_token_id,
         sample_image_size,
@@ -54,7 +53,6 @@ class CheckpointerBase:
     ):
         self.datamodule = datamodule
         self.output_dir = output_dir
-        self.instance_identifier = instance_identifier
         self.placeholder_token = placeholder_token
         self.placeholder_token_id = placeholder_token_id
         self.sample_image_size = sample_image_size
@@ -90,7 +88,7 @@ class CheckpointerBase:
                     if j * data.batch_size < self.sample_batch_size * self.sample_batches
                 ]
                 prompts = [
-                    prompt.format(identifier=self.instance_identifier)
+                    prompt
                     for batch in batches
                     for prompt in batch["prompts"]
                 ]