Update _utils.py
This commit is contained in:
parent
2997923ce5
commit
473665b107
1 changed files with 1 additions and 1 deletions
|
|
@ -1667,7 +1667,7 @@ except:
|
|||
@dataclass
|
||||
class TorchAOConfig:
|
||||
qat_scheme : str = "int4"
|
||||
base_config : AOBaseConfig = Int4WeightOnlyConfig
|
||||
base_config : AOBaseConfig = Int4WeightOnlyConfig(group_size = 128)
|
||||
filter_fn : Callable = None
|
||||
group_size : int = 128
|
||||
pass
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue