mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
fix fp16 mm is selected instead of fp8 mm
This commit is contained in:
@@ -245,7 +245,7 @@ def sdnq_quantize_layer_weight(weight, layer_class_name=None, weights_dtype="int
|
||||
if quantized_matmul_dtype is None:
|
||||
if dtype_dict[weights_dtype]["is_integer"]:
|
||||
quantized_matmul_dtype = "int8"
|
||||
elif dtype_dict[weights_dtype]["num_bits"] == 8:
|
||||
elif dtype_dict[weights_dtype]["num_bits"] <= 12:
|
||||
quantized_matmul_dtype = "float8_e4m3fn"
|
||||
else:
|
||||
quantized_matmul_dtype = "float16"
|
||||
|
||||
Reference in New Issue
Block a user