-
Notifications
You must be signed in to change notification settings - Fork 862
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
[core] move transformers to setup_utils, bump dependency version #3984
[core] move transformers to setup_utils, bump dependency version #3984
Conversation
In Chronos, we would like to depend on Currently, bumping the Looking forward to your comments. |
@prateekdesai04 we should test this change with torch update |
timeseries/setup.py
Outdated
@@ -55,7 +57,9 @@ | |||
], | |||
} | |||
|
|||
extras_require["all"] = [] | |||
extras_require["all"] = [ | |||
"optimum[openvino,onnxruntime,nncf]>=1.17,<1.18", # for faster CPU inference in pretrained models |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do we need neuron suppport to speed up inference on Inf2 architecture?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not at the moment, though this will come in v1.2.
@tonyhoo Just pushed the latest Torch + CUDA image, running this PR #3982 |
Job PR-3984-a4b7229 is done. |
Converted to draft, pending the PyTorch upgrade. and thanks @zhiqiangdon ! |
a4b7229
to
a0635c5
Compare
|
Job PR-3984-a0635c5 is done. |
|
@zhiqiangdon for your review of test fails |
|
caf620e
to
314e8c3
Compare
|
Job PR-3984-314e8c3 is done. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
…tch-4 * 'master' of https://github.com/awslabs/autogluon: (46 commits) [core] move transformers to setup_utils, bump dependency version (autogluon#3984) [AutoMM] Fix one lightning upgrade issue (autogluon#3991) [CI][Feature] Create a package version table (autogluon#3972) [v.1.1][Upgrade] PyTorch 2.1 and CUDA 12.1 upgrade (autogluon#3982) [WIP] Code implementation of Conv-LoRA (autogluon#3933) [timeseries] Ensure that all metrics handle missing values in the target (autogluon#3966) [timeseries] Fix path and device bugs (autogluon#3979) [AutoMM]Remove grounding-dino (autogluon#3974) [Docs] Update install modules content (autogluon#3976) Add note on pd.to_datetime (autogluon#3975) [AutoMM] Improve DINO performance (autogluon#3970) Minor correction in differ to pick correct environment (autogluon#3968) Fix windows python 3.11 issue by removing ray (autogluon#3956) [CI][Feature] Package Version Comparator (autogluon#3962) [timeseries] Add support for categorical covariates (autogluon#3874) [timeseries] Add method for plotting forecasts (autogluon#3889) Update conf.py copyright to reflect current year (autogluon#3932) [Timeseries][CI]Refactor CI to skip AutoMM and Tabular tests w.r.t timeseries changes (autogluon#3942) Fix HPO crash in memory check (autogluon#3931) [AutoMM][CI] Capping scikit-learn to avoid HPO test failure (autogluon#3947) ...
…ogluon#3984) Co-authored-by: Zhiqiang Tang <zhiqiang.tang@rutgers.edu>
…ogluon#3984) Co-authored-by: Zhiqiang Tang <zhiqiang.tang@rutgers.edu>
Issue #, if available:
Description of changes:
Chronos implementation (cf #3978) requires
transformers>=4.36
. This PR moves transformers and accelerate version ranges to_setup_utils.py
and bumps the version.By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.