diff options
Diffstat (limited to 'training/strategy/dreambooth.py')
| -rw-r--r-- | training/strategy/dreambooth.py | 2 |
1 files changed, 1 insertions, 1 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, |