diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-21 21:13:53 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-21 21:13:53 +0000 |
| commit | 3366e494a1147e570d8527eea19da88edb3a1e0c (patch) | |
| tree | 62c196b5c13a337f0bf72aca3a07d1c51b7f925f /modules/sd_models.py | |
| parent | 8faac8b96313c6c4bf0a81bddecff4d6ba22ac25 (diff) | |
| download | stable-diffusion-webui-gfx803-3366e494a1147e570d8527eea19da88edb3a1e0c.tar.gz stable-diffusion-webui-gfx803-3366e494a1147e570d8527eea19da88edb3a1e0c.tar.bz2 stable-diffusion-webui-gfx803-3366e494a1147e570d8527eea19da88edb3a1e0c.zip | |
option to pad prompt/neg prompt to be same length
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index b1afbaa7..91b3eb11 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -508,6 +508,11 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None): timer.record("scripts callbacks")
+ with devices.autocast(), torch.no_grad():
+ sd_model.cond_stage_model_empty_prompt = sd_model.cond_stage_model([""])
+
+ timer.record("calculate empty prompt")
+
print(f"Model loaded in {timer.summary()}.")
return sd_model
|