diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-24 10:25:58 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-24 10:25:58 +0100 |
| commit | e09aaedd0e74f2fc6e2a53f233914803c65e127c (patch) | |
| tree | 186a6442cb4de3210837ca459aad81a22a3f37ee /training | |
| parent | Update (diff) | |
| download | textual-inversion-diff-e09aaedd0e74f2fc6e2a53f233914803c65e127c.tar.gz textual-inversion-diff-e09aaedd0e74f2fc6e2a53f233914803c65e127c.tar.bz2 textual-inversion-diff-e09aaedd0e74f2fc6e2a53f233914803c65e127c.zip | |
Training update
Diffstat (limited to 'training')
| -rw-r--r-- | training/ti.py | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/training/ti.py b/training/ti.py index a5e407b..8b2fdd6 100644 --- a/training/ti.py +++ b/training/ti.py | |||
| @@ -18,8 +18,6 @@ def patch_trainable_embeddings(text_encoder: CLIPTextModel, new_ids: list[int]): | |||
| 18 | 18 | ||
| 19 | text_encoder.text_model.embeddings = text_embeddings | 19 | text_encoder.text_model.embeddings = text_embeddings |
| 20 | 20 | ||
| 21 | return text_embeddings | ||
| 22 | |||
| 23 | 21 | ||
| 24 | class TrainableEmbeddings(CLIPTextEmbeddings): | 22 | class TrainableEmbeddings(CLIPTextEmbeddings): |
| 25 | def __init__(self, config: CLIPTextConfig, new_ids: list[int]): | 23 | def __init__(self, config: CLIPTextConfig, new_ids: list[int]): |