Skip to content

Commit 5db136c

Browse files
committed
Fix segmentation fault
1 parent 46ea733 commit 5db136c

File tree

1 file changed

+3
-2
lines changed
  • examples/models/llama3_2_vision/runner

1 file changed

+3
-2
lines changed

examples/models/llama3_2_vision/runner/native.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,10 @@ def __init__(self, args):
4444
use_kv_cache=args.kv_cache,
4545
vocab_size=params["vocab_size"],
4646
)
47+
self.model_bytes = None
4748
with open(args.pte, "rb") as f:
48-
model_bytes = f.read()
49-
self.model = _load_for_executorch_from_buffer(model_bytes)
49+
self.model_bytes = f.read()
50+
self.model = _load_for_executorch_from_buffer(self.model_bytes)
5051
# self.model = _load_for_executorch(args.pte)
5152
self.use_kv_cache = args.kv_cache
5253

0 commit comments

Comments
 (0)