Skip to content

Commit 8b0c9a5

Browse files
committed
Rebase fix
1 parent 9c41e1b commit 8b0c9a5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/llama3_2_vision/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def __init__(self, **kwargs):
4747
self.args = kwargs.get("args", None)
4848

4949

50-
ckpt_dir = get_default_model_resource_dir()
50+
ckpt_dir = get_default_model_resource_dir(__file__)
5151
# Single checkpoint file.
5252
checkpoint_path = kwargs.get("checkpoint", ckpt_dir / "demo_rand_params.pth")
5353
# Sharded checkpoint.

0 commit comments

Comments
 (0)