diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-24 11:57:59 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-24 11:57:59 +0300 |
| commit | 2c11e9009ea18bab4ce2963d44db0c6fd3227370 (patch) | |
| tree | cf70006b4d1d6df1f42ea944416b1034ae32a92b /modules/sd_hijack_open_clip.py | |
| parent | 1f26815dd3ea5a0b8c094854e47219aec866f28c (diff) | |
repair --medvram for SD2.x too after SDXL update
Diffstat (limited to 'modules/sd_hijack_open_clip.py')
| -rw-r--r-- | modules/sd_hijack_open_clip.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_hijack_open_clip.py b/modules/sd_hijack_open_clip.py index bb0b96c7..25c5e983 100644 --- a/modules/sd_hijack_open_clip.py +++ b/modules/sd_hijack_open_clip.py @@ -32,7 +32,7 @@ class FrozenOpenCLIPEmbedderWithCustomWords(sd_hijack_clip.FrozenCLIPEmbedderWit def encode_embedding_init_text(self, init_text, nvpt):
ids = tokenizer.encode(init_text)
ids = torch.asarray([ids], device=devices.device, dtype=torch.int)
- embedded = self.wrapped.model.token_embedding.wrapped(ids.to(self.wrapped.model.token_embedding.wrapped.weight.device)).squeeze(0)
+ embedded = self.wrapped.model.token_embedding.wrapped(ids).squeeze(0)
return embedded
|