diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-11-25 12:35:09 +0800 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-11-25 12:35:09 +0800 |
| commit | 40ac134c553ac824d4a96666bba14d550300daa5 (patch) | |
| tree | 899aa6aadafc3e4a0ffc969b508460d1da27f2d5 /modules/sd_models.py | |
| parent | f5d719d1f1baa775d838aa75d9af1971bcc78e8f (diff) | |
Fix pre-fp8
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 0a7777f1..90437c87 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -357,7 +357,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer sd_model_hash = checkpoint_info.calculate_shorthash()
timer.record("calculate hash")
- if not check_fp8(model) and devices.fp8:
+ if devices.fp8:
# prevent model to load state dict in fp8
model.half()
|