1 files changed, 1 insertions, 1 deletions

diff --git a/training/strategy/dreambooth.py b/training/strategy/dreambooth.py
index 695174a..42624cd 100644
--- a/training/strategy/dreambooth.py
+++ b/training/strategy/dreambooth.py
@@ -198,7 +198,7 @@ def dreambooth_prepare(
 
     text_encoder.text_model.embeddings.requires_grad_(False)
 
-    return text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler, {}
+    return text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler
 
 
 dreambooth_strategy = TrainingStrategy(