diff options
| author | Ritesh Gangnani <riteshgangnani10> | 2023-11-05 15:01:57 +0000 |
|---|---|---|
| committer | Ritesh Gangnani <riteshgangnani10> | 2023-11-05 15:01:57 +0000 |
| commit | 44c5097375ae4cf40300c09473bb46cf6c5d6cb7 (patch) | |
| tree | 1dd13c7a0413d8b1ef64167e0265242208eeb692 /modules/sd_models.py | |
| parent | 44db35fb1ad5d07837e890a0fd3c00addfb0402c (diff) | |
| download | stable-diffusion-webui-gfx803-44c5097375ae4cf40300c09473bb46cf6c5d6cb7.tar.gz stable-diffusion-webui-gfx803-44c5097375ae4cf40300c09473bb46cf6c5d6cb7.tar.bz2 stable-diffusion-webui-gfx803-44c5097375ae4cf40300c09473bb46cf6c5d6cb7.zip | |
Use devices.torch_gc() instead of empty_cache()
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index ef96d29d..2242c363 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -347,7 +347,6 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer model.is_sd2 = not model.is_sdxl and hasattr(model.cond_stage_model, 'model')
model.is_sd1 = not model.is_sdxl and not model.is_sd2
model.is_ssd = model.is_sdxl and 'model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight' not in state_dict.keys()
-
if model.is_sdxl:
sd_models_xl.extend_sdxl(model)
|