Fix mainline

winskuo-quic · winskuo-quic · commit 43ce5b25a47b · 2025-08-03T01:35:53.000+08:00
diff --git a/examples/qualcomm/oss_scripts/llama/llama.py b/examples/qualcomm/oss_scripts/llama/llama.py
@@ -595,7 +595,7 @@ def permute(w, heads):
                 annotate_conv=args.ptq != "16a8w",
             ),
         )
-        if args.decoder_model == {"stories110m", "stories260k"}:
+        if args.decoder_model in {"stories110m", "stories260k"}:
             custom_annotations = custom_annotations + (
                 annotate_linear_16a8w_in_affine_layer,
             )

Original file line number	Diff line number	Diff line change
`@@ -595,7 +595,7 @@ def permute(w, heads):`
`595`	`595`	`annotate_conv=args.ptq != "16a8w",`
`596`	`596`	`),`
`597`	`597`	`)`
`598`		`- if args.decoder_model == {"stories110m", "stories260k"}:`
	`598`	`+ if args.decoder_model in {"stories110m", "stories260k"}:`
`599`	`599`	`custom_annotations = custom_annotations + (`
`600`	`600`	`annotate_linear_16a8w_in_affine_layer,`
`601`	`601`	`)`