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

Support file paths for ONNX predictor #1711

Merged
merged 31 commits into from
Dec 22, 2020
Merged

Conversation

RobertLucian
Copy link
Member

@RobertLucian RobertLucian commented Dec 18, 2020

Closes #1686.

Blocked on #1702.


checklist:

  • run make test and make lint
  • test manually (i.e. build/push all images, restart operator, and re-deploy APIs)

@RobertLucian RobertLucian requested a review from deliahu December 18, 2020 00:54
@RobertLucian RobertLucian changed the base branch from master to improvement/python-api-spec December 18, 2020 00:54
if api.Predictor.Models.Path != nil && strings.HasSuffix(*api.Predictor.Models.Path, ".onnx") {
downloadContainerArs = append(downloadContainerArs, downloadContainerArg{
From: *api.Predictor.Models.Path,
To: path.Join(_modelDir, consts.SingleModelName, "1"),
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: I think if we add ItemName: "the onnx model" to the struct, it will print a nice message in the downloader container

Copy link
Member Author

@RobertLucian RobertLucian Dec 22, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thought of not doing it because adding this log would have been misleading since we advertise the predictor.models predictor as having the LR active. That being said, I actually think this is a good idea because it highlights to the user that a model gets downloaded at deploy time (suggestive that it's a special case and LR is not active) and it's also potentially easier for us to debug if the user shares their logs.

Base automatically changed from improvement/python-api-spec to master December 22, 2020 00:51
# Conflicts:
#	cli/local/docker_spec.go
#	cli/local/model_cache.go
#	docs/workloads/batch/configuration.md
#	docs/workloads/realtime/configuration.md
#	docs/workloads/realtime/models.md
#	docs/workloads/realtime/predictors.md
#	pkg/cortex/serve/cortex_internal/lib/model/type.py
#	pkg/operator/operator/k8s.go
#	pkg/types/spec/errors.go
#	pkg/types/spec/validations.go
#	pkg/types/userconfig/api.go
@RobertLucian RobertLucian merged commit dcaba64 into master Dec 22, 2020
@RobertLucian RobertLucian deleted the feature/single-model-paths branch December 22, 2020 02:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Support file paths for ONNX predictor
3 participants