Skip to content

Commit ececb79

Browse files
committed
fix: reformatted the argument file
Signed-off-by: omobayode.fagbohungbe <omobayode.fagbohungbe@ibm.com>
1 parent 26330a4 commit ececb79

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

fms_mo/training_args.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,12 +207,11 @@ class GPTQArguments(TypeChecker):
207207
autotune_warmup_after_quantized: bool = False
208208
cache_examples_on_gpu: bool = True
209209
use_version2: bool = False
210-
v2_mem_device: Optional[str] = field(
210+
v2_mem_device: str = field(
211211
default="cpu", metadata={"choices": ["auto", "cpu", "cuda"]}
212212
)
213213

214214

215-
216215
@dataclass
217216
class FP8Arguments(TypeChecker):
218217
"""Dataclass for FP8 related arguments that will be used by llm-compressor."""

0 commit comments

Comments
 (0)