Commit 40ac134c authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf
Browse files

Fix pre-fp8

parent f5d719d1

modules/sd_models.py

+1 −1
Original line number Diff line number Diff line
@@ -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()

CRA Git | Maintained and supported by SUSTech CRA and CCSE