Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[SageMaker] Fix download location for model inference #1132

Merged
merged 3 commits into from
Jan 14, 2025

Conversation

thvasilo
Copy link
Contributor

Issue #, if available:

Description of changes:

  • Remove double download of model files and ensure models are downloaded under /tmp
  • Tested by running a node classification job with ogb-arxiv data

Fixes #949

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@thvasilo thvasilo added this to the 0.4 release milestone Jan 12, 2025
@thvasilo thvasilo self-assigned this Jan 12, 2025
sagemaker/launch/launch_train.py Show resolved Hide resolved
sagemaker/launch/launch_infer.py Show resolved Hide resolved
@thvasilo thvasilo merged commit 6c0f81e into awslabs:main Jan 14, 2025
8 checks passed
jalencato pushed a commit that referenced this pull request Jan 14, 2025
*Issue #, if available:*

*Description of changes:*

* Remove double download of model files and ensure models are downloaded
under /tmp
* Tested by running a node classification job with ogb-arxiv data

Fixes #949

By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

SageMaker inference download locations are misconfigured, and models are downloaded twice
3 participants