diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-14 10:27:50 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-14 10:27:50 +0000 |
| commit | 1a43524018ea3e64b93be2abc2a49b6159515442 (patch) | |
| tree | 4d2e3a736558cf6b1bc65bf969c1652c6a5d2f8f /modules/sd_hijack.py | |
| parent | 5f5435eb1aafbd2dfda3e00024d8528addd17e03 (diff) | |
| download | stable-diffusion-webui-gfx803-1a43524018ea3e64b93be2abc2a49b6159515442.tar.gz stable-diffusion-webui-gfx803-1a43524018ea3e64b93be2abc2a49b6159515442.tar.bz2 stable-diffusion-webui-gfx803-1a43524018ea3e64b93be2abc2a49b6159515442.zip | |
fix model loading twice in some situations
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 7e50f1ab..14e7f799 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -216,6 +216,9 @@ class StableDiffusionModelHijack: self.comments = []
def get_prompt_lengths(self, text):
+ if self.clip is None:
+ return "-", "-"
+
_, token_count = self.clip.process_texts([text])
return token_count, self.clip.get_target_prompt_token_count(token_count)
|