From 1bd386f98bb076fe62696808e02a9bd9b9b64b42 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Fri, 23 Dec 2022 21:47:12 +0100
Subject: Improved class prompt handling

---
 training/util.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

(limited to 'training')

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"]
                 ]
-- 
cgit v1.2.3-70-g09d2