OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-01 03:03:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix pre-fp8

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2023-11-25 12:35:09 +08:00
parent f5d719d1f1
commit 40ac134c55
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models.py
View File

@ -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()