diff options
| author | Volpeon <git@volpeon.ink> | 2023-05-16 07:12:14 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-05-16 07:12:14 +0200 |
| commit | b31fcb741432076f7e2f3ec9423ad935a08c6671 (patch) | |
| tree | 2ab052d3bd617a56c4ea388c200da52cff39ba37 /models/clip/embeddings.py | |
| parent | Fix for latest PEFT (diff) | |
| download | textual-inversion-diff-b31fcb741432076f7e2f3ec9423ad935a08c6671.tar.gz textual-inversion-diff-b31fcb741432076f7e2f3ec9423ad935a08c6671.tar.bz2 textual-inversion-diff-b31fcb741432076f7e2f3ec9423ad935a08c6671.zip | |
Support LoRA training for token embeddings
Diffstat (limited to 'models/clip/embeddings.py')
| -rw-r--r-- | models/clip/embeddings.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/models/clip/embeddings.py b/models/clip/embeddings.py index 2b23bd3..7c7f2ac 100644 --- a/models/clip/embeddings.py +++ b/models/clip/embeddings.py | |||
| @@ -86,6 +86,9 @@ def patch_managed_embeddings( | |||
| 86 | alpha: int = 8, | 86 | alpha: int = 8, |
| 87 | dropout: float = 0.0 | 87 | dropout: float = 0.0 |
| 88 | ) -> ManagedCLIPTextEmbeddings: | 88 | ) -> ManagedCLIPTextEmbeddings: |
| 89 | if isinstance(text_encoder.text_model.embeddings, ManagedCLIPTextEmbeddings): | ||
| 90 | return text_encoder.text_model.embeddings | ||
| 91 | |||
| 89 | text_embeddings = ManagedCLIPTextEmbeddings( | 92 | text_embeddings = ManagedCLIPTextEmbeddings( |
| 90 | text_encoder.config, | 93 | text_encoder.config, |
| 91 | text_encoder.text_model.embeddings, | 94 | text_encoder.text_model.embeddings, |