From 300deaa789a0321f32d5e7f04d9860eaa258110e Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 4 Oct 2022 19:22:22 +0200
Subject: Add Textual Inversion with class dataset (a la Dreambooth)

---
 data/dreambooth/prompt.py | 18 ------------------
 1 file changed, 18 deletions(-)
 delete mode 100644 data/dreambooth/prompt.py

(limited to 'data/dreambooth/prompt.py')

diff --git a/data/dreambooth/prompt.py b/data/dreambooth/prompt.py
deleted file mode 100644
index b3a83ce..0000000
--- a/data/dreambooth/prompt.py
+++ /dev/null
@@ -1,18 +0,0 @@
-from torch.utils.data import Dataset
-
-
-class PromptDataset(Dataset):
-    def __init__(self, prompt, nprompt, num_samples):
-        self.prompt = prompt
-        self.nprompt = nprompt
-        self.num_samples = num_samples
-
-    def __len__(self):
-        return self.num_samples
-
-    def __getitem__(self, index):
-        example = {}
-        example["prompt"] = self.prompt
-        example["nprompt"] = self.nprompt
-        example["index"] = index
-        return example
-- 
cgit v1.2.3-54-g00ecf