Skip to content

S
Stable Diffusion Webui
Commit 40ac134c authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf
Browse files
Options

Fix pre-fp8

parent f5d719d1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/sd_models.py
View file @ 40ac134c
...@@ -357,7 +357,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer ...@@ -357,7 +357,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
sd_model_hash = checkpoint_info.calculate_shorthash() sd_model_hash = checkpoint_info.calculate_shorthash()
timer.record("calculate hash") timer.record("calculate hash")
if not check_fp8(model) and devices.fp8: if devices.fp8:
# prevent model to load state dict in fp8 # prevent model to load state dict in fp8
model.half() model.half()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment