Skip to content

Commit b319eae

Browse files
committed
chore: precision -> model_precision option
1 parent eeef1c4 commit b319eae

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/py/dynamo/llm/test_llm_models.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,15 +23,15 @@ def test_llm_decoder_layer(precision):
2323
args.debug = False
2424
args.num_tokens = 128
2525
args.model = "Qwen/Qwen2.5-0.5B-Instruct"
26-
args.precision = precision
26+
args.model_precision = precision
2727
args.min_block_size = 1
2828
args.prompt = "What is parallel programming ?"
29-
if args.precision == "FP16":
29+
if args.model_precision == "FP16":
3030
dtype = torch.float16
31-
elif args.precision == "BF16":
31+
elif args.model_precision == "BF16":
3232
dtype = torch.bfloat16
3333
else:
34-
args.precision = "FP32"
34+
args.model_precision = "FP32"
3535
dtype = torch.float32
3636

3737
model = (

0 commit comments

Comments
 (0)