-
Notifications
You must be signed in to change notification settings - Fork 8
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
Add ChronosModel
and ChronosBoltModel
#511
Conversation
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
🚀 Deployed on https://deploy-preview-511--etna-docs.netlify.app |
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## master #511 +/- ##
==========================================
- Coverage 90.53% 90.32% -0.22%
==========================================
Files 247 256 +9
Lines 16568 17199 +631
==========================================
+ Hits 15000 15535 +535
- Misses 1568 1664 +96 ☔ View full report in Codecov by Sentry. |
During the first initialization model is downloaded from huggingface and saved to local ``cache_dir``. | ||
All following initializations model will be loaded from ``cache_dir``. See ``pretrained_model_name_or_path`` parameter of :py:func:`transformers.PreTrainedModel.from_pretrained` | ||
- If local path, model will not be saved to local ``cache_dir``. | ||
- If external url, it must be zip archive with the same name as model directory inside. Model will be downloaded to ``cache_dir``. |
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.
Are we checking it somehow?
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.
No. I think real check should be complex like checking ***.safetensor
or pytorch_model.bin
or config.json
is in directory. As the main purpose is developer cases, so we can pay less attention to it.
View / edit / reply to this conversation on ReviewNB d-a-bunin commented on 2024-12-09T08:31:25Z Line #3. model_chronos = ChronosModel(path_or_url="amazon/chronos-t5-tiny", encoder_length=2 * HORIZON, num_samples=10) Do I understand correctly that we set encoder_length to 2*HORIZON only to demonstrate how this value works in the next cell? egoriyaa commented on 2024-12-09T10:46:43Z Yes. Also |
Yes. Also View entire conversation on ReviewNB |
Before submitting (must do checklist)
Proposed Changes
Closing issues
closes #509