diff options
author | Volpeon <git@volpeon.ink> | 2023-01-16 10:31:55 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-01-16 10:31:55 +0100 |
commit | 89afcfda3f824cc44221e877182348f9b09687d2 (patch) | |
tree | 804b84322e5caa8fb861322ce6970bef4b532c61 /training/strategy | |
parent | Extended Dreambooth: Train TI tokens separately (diff) | |
download | textual-inversion-diff-89afcfda3f824cc44221e877182348f9b09687d2.tar.gz textual-inversion-diff-89afcfda3f824cc44221e877182348f9b09687d2.tar.bz2 textual-inversion-diff-89afcfda3f824cc44221e877182348f9b09687d2.zip |
Handle empty validation dataset
Diffstat (limited to 'training/strategy')
-rw-r--r-- | training/strategy/dreambooth.py | 2 | ||||
-rw-r--r-- | training/strategy/ti.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/training/strategy/dreambooth.py b/training/strategy/dreambooth.py index 6e7ebe2..aeaa828 100644 --- a/training/strategy/dreambooth.py +++ b/training/strategy/dreambooth.py | |||
@@ -26,7 +26,7 @@ def dreambooth_strategy( | |||
26 | vae: AutoencoderKL, | 26 | vae: AutoencoderKL, |
27 | sample_scheduler: DPMSolverMultistepScheduler, | 27 | sample_scheduler: DPMSolverMultistepScheduler, |
28 | train_dataloader: DataLoader, | 28 | train_dataloader: DataLoader, |
29 | val_dataloader: DataLoader, | 29 | val_dataloader: Optional[DataLoader], |
30 | output_dir: Path, | 30 | output_dir: Path, |
31 | seed: int, | 31 | seed: int, |
32 | train_text_encoder_epochs: int, | 32 | train_text_encoder_epochs: int, |
diff --git a/training/strategy/ti.py b/training/strategy/ti.py index 753dce0..568f9eb 100644 --- a/training/strategy/ti.py +++ b/training/strategy/ti.py | |||
@@ -26,7 +26,7 @@ def textual_inversion_strategy( | |||
26 | vae: AutoencoderKL, | 26 | vae: AutoencoderKL, |
27 | sample_scheduler: DPMSolverMultistepScheduler, | 27 | sample_scheduler: DPMSolverMultistepScheduler, |
28 | train_dataloader: DataLoader, | 28 | train_dataloader: DataLoader, |
29 | val_dataloader: DataLoader, | 29 | val_dataloader: Optional[DataLoader], |
30 | output_dir: Path, | 30 | output_dir: Path, |
31 | seed: int, | 31 | seed: int, |
32 | placeholder_tokens: list[str], | 32 | placeholder_tokens: list[str], |