Skip to content

Commit bf51fdc

Browse files
add support for < 8 bits in weight_type
not tested end-to-end yet.
1 parent 67113b3 commit bf51fdc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/NeurekaTestConf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def check_valid_out_type(cls, v: IntegerType) -> IntegerType:
6565
@field_validator("weight_type")
6666
@classmethod
6767
def check_valid_weight_type(cls, v: IntegerType) -> IntegerType:
68-
NeurekaTestConf._check_type("weight_type", v, ["int8"])
68+
NeurekaTestConf._check_type("weight_type", v, ["int8", "int7", "int6", "int5", "int4", "int3", "int2"])
6969
return v
7070

7171
@field_validator("scale_type")

0 commit comments

Comments
 (0)