We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9c41e1b commit 8b0c9a5Copy full SHA for 8b0c9a5
examples/models/llama3_2_vision/model.py
@@ -47,7 +47,7 @@ def __init__(self, **kwargs):
47
self.args = kwargs.get("args", None)
48
49
50
- ckpt_dir = get_default_model_resource_dir()
+ ckpt_dir = get_default_model_resource_dir(__file__)
51
# Single checkpoint file.
52
checkpoint_path = kwargs.get("checkpoint", ckpt_dir / "demo_rand_params.pth")
53
# Sharded checkpoint.
0 commit comments