Unverified Commit 08d4fb6e authored by Pierre LeMoine's avatar Pierre LeMoine Committed by GitHub
Browse files

[dreambooth] Using already created `Path` in dataset (#681)

using already created `Path` in dataset
parent a8a3a20d
......@@ -244,7 +244,7 @@ class DreamBoothDataset(Dataset):
if class_data_root is not None:
self.class_data_root = Path(class_data_root)
self.class_data_root.mkdir(parents=True, exist_ok=True)
self.class_images_path = list(Path(class_data_root).iterdir())
self.class_images_path = list(self.class_data_root.iterdir())
self.num_class_images = len(self.class_images_path)
self._length = max(self.num_class_images, self.num_instance_images)
self.class_prompt = class_prompt
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment