convert : fix flake8 lint
This commit is contained in:
parent
987862ad8c
commit
3770d9410d
|
|
@ -416,7 +416,7 @@ class ModelBase:
|
||||||
block_size = weight_config.get("block_structure", None)
|
block_size = weight_config.get("block_structure", None)
|
||||||
strategy = weight_config.get("strategy")
|
strategy = weight_config.get("strategy")
|
||||||
assert strategy == "channel" or strategy == "block"
|
assert strategy == "channel" or strategy == "block"
|
||||||
assert weight_config.get("group_size") == None # didn't find a model using this yet
|
assert weight_config.get("group_size") is None # didn't find a model using this yet
|
||||||
for name in self.model_tensors.keys():
|
for name in self.model_tensors.keys():
|
||||||
if name.endswith(".weight_scale"):
|
if name.endswith(".weight_scale"):
|
||||||
weight_name = name.removesuffix("_scale")
|
weight_name = name.removesuffix("_scale")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue