Skip to content

Commit f77226d

Browse files
committed
use iterable
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 9298e82 commit f77226d

File tree

1 file changed

+2
-2
lines changed
  • src/llmcompressor/modifiers/transform/spinquant

1 file changed

+2
-2
lines changed

src/llmcompressor/modifiers/transform/spinquant/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Optional, List, Literal
1+
from typing import Optional, List, Literal, Iterable
22

33
from compressed_tensors.transform import TransformConfig, TransformScheme, TransformArgs, apply_transform_config
44
from pydantic import BaseModel, field_validator, Field
@@ -77,7 +77,7 @@ class SpinquantRotation(Enum):
7777
R4 = "R4"
7878

7979
class SpinQuantModifier(Modifier):
80-
rotations: List[SpinquantRotation] = Field(default_factory=lambda: ["R1", "R2"])
80+
rotations: Iterable[SpinquantRotation] = ("R1", "R2")
8181
transform_type: Literal["hadamard", "random-hadamard", "random-matrix"] = Field(default="hadamard")
8282
randomize: bool = Field(default=False)
8383
learnable: bool = Field(default=False)

0 commit comments

Comments
 (0)