diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-25 14:59:00 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-25 14:59:00 +0100 |
| commit | 1af6c15f795b5ba4df9179d8c59c6b595040a33f (patch) | |
| tree | fa7c033a6c259b64fa84b5483894150b07c9337f /train_dreambooth.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-1af6c15f795b5ba4df9179d8c59c6b595040a33f.tar.gz textual-inversion-diff-1af6c15f795b5ba4df9179d8c59c6b595040a33f.tar.bz2 textual-inversion-diff-1af6c15f795b5ba4df9179d8c59c6b595040a33f.zip | |
Update
Diffstat (limited to 'train_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 1 |
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(): | |||
| 556 | text_encoder.resize_token_embeddings(len(tokenizer)) | 556 | text_encoder.resize_token_embeddings(len(tokenizer)) |
| 557 | 557 | ||
| 558 | token_embeds = text_encoder.get_input_embeddings().weight.data | 558 | token_embeds = text_encoder.get_input_embeddings().weight.data |
| 559 | original_token_embeds = token_embeds.clone().to(accelerator.device) | ||
| 560 | initializer_token_embeddings = text_encoder.get_input_embeddings()(initializer_token_ids) | 559 | initializer_token_embeddings = text_encoder.get_input_embeddings()(initializer_token_ids) |
| 561 | 560 | ||
| 562 | for (token_id, embeddings) in zip(placeholder_token_id, initializer_token_embeddings): | 561 | for (token_id, embeddings) in zip(placeholder_token_id, initializer_token_embeddings): |