From 1af6c15f795b5ba4df9179d8c59c6b595040a33f Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 25 Dec 2022 14:59:00 +0100 Subject: Update --- train_dreambooth.py | 1 - 1 file changed, 1 deletion(-) (limited to 'train_dreambooth.py') 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): -- cgit v1.2.3-54-g00ecf