diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py index f8d901671..859aa603c 100644 --- a/modules/sdnq/common.py +++ b/modules/sdnq/common.py @@ -124,7 +124,7 @@ dtype_dict["bf16"] = dtype_dict["bfloat16"] dtype_dict["fp16"] = dtype_dict["float16"] dtype_dict["fp8"] = dtype_dict["float8_e4m3fn"] dtype_dict["fp7"] = dtype_dict["float7_e3m3fn"] -dtype_dict["fp6"] = dtype_dict["float6_e2m3fn"] +dtype_dict["fp6"] = dtype_dict["float6_e3m2fn"] dtype_dict["fp5"] = dtype_dict["float5_e2m2fn"] dtype_dict["fp4"] = dtype_dict["float4_e2m1fn"] dtype_dict["fp3"] = dtype_dict["float3_e1m1fn"] diff --git a/modules/shared.py b/modules/shared.py index b8844984c..9d6bc157b 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -70,7 +70,7 @@ restricted_opts = { } resize_modes = ["None", "Fixed", "Crop", "Fill", "Outpaint", "Context aware"] max_workers = 12 -sdnq_quant_modes = ["int8", "int7", "int6", "uint5", "uint4", "uint3", "uint2", "float8_e4m3fn", "float7_e3m3fn", "float6_e2m3fn", "float5_e2m2fn", "float4_e2m1fn", "float3_e1m1fn", "float2_e1m0fn"] +sdnq_quant_modes = ["int8", "int7", "int6", "uint5", "uint4", "uint3", "uint2", "float8_e4m3fn", "float7_e3m3fn", "float6_e3m2fn", "float5_e2m2fn", "float4_e2m1fn", "float3_e1m1fn", "float2_e1m0fn"] sdnq_matmul_modes = ["auto", "int8", "float8_e4m3fn", "float16"] default_hfcache_dir = os.environ.get("SD_HFCACHEDIR", None) or os.path.join(paths.models_path, 'huggingface') state = shared_state.State()