1 files changed, 0 insertions, 1 deletions

diff --git a/train_dreambooth.py b/train_dreambooth.py
index 8cb6414..e239833 100644
--- a/train_dreambooth.py
+++ b/train_dreambooth.py
@@ -556,7 +556,6 @@ def main():
         text_encoder.resize_token_embeddings(len(tokenizer))
 
         token_embeds = text_encoder.get_input_embeddings().weight.data
-        original_token_embeds = token_embeds.clone().to(accelerator.device)
         initializer_token_embeddings = text_encoder.get_input_embeddings()(initializer_token_ids)
 
         for (token_id, embeddings) in zip(placeholder_token_id, initializer_token_embeddings):