Skip to content

Commit 1fc6820

Browse files
authored
Merge pull request #380 from jondea/fix-vision-instruct-using-version-1
pytorch: fix quality of vision instruct by using version=1
2 parents 3039334 + 294a66b commit 1fc6820

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ML-Frameworks/pytorch-aarch64/examples/llama_vision_instruct.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def main(args):
5353
layout=layout,
5454
weight_dtype=torch.int4,
5555
intx_packing_format="opaque_aten_kleidiai",
56+
version=1,
5657
),
5758
)
5859

0 commit comments

Comments
 (0)