Skip to content

Commit 1d97568

Browse files
committed
Replace str with enum representation
1 parent fd81a2a commit 1d97568

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/quantization_w4a8/gpt_oss_quantization_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ def preprocess(example):
238238
ds = ds.map(preprocess)
239239

240240
# Tokenize for GPTQ (required for GPTQ, optional for others)
241-
if args.algorithm == "gptq":
241+
if args.algorithm == QuantizationAlgorithm.GPTQ:
242242

243243
def tokenize(sample):
244244
return tokenizer(

0 commit comments

Comments
 (0)