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

Add models to demo docker image #1978

Merged
merged 2 commits into from
Jan 11, 2022
Merged

Add models to demo docker image #1978

merged 2 commits into from
Jan 11, 2022

Conversation

ZanSara
Copy link
Contributor

@ZanSara ZanSara commented Jan 10, 2022

Introduces a small utility to download and cache the demo's model and NLTK data, then modifies Dockerimage and Dockerimage-GPU to call this function at build time.

This will provide users with a demo that does not need to download any model at its first run, making the demo installation process easier and a bit faster too.

Closes #1652

Copy link
Member

@julian-risch julian-risch left a comment

Choose a reason for hiding this comment

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

While the model caching works for me, it seems that the nltk caching does not. Could you please doublecheck that? Otherwise looks good to me! 👍

haystack/utils/docker.py Outdated Show resolved Hide resolved
@ZanSara
Copy link
Contributor Author

ZanSara commented Jan 11, 2022

I don't see anymore logs related to punkt when starting the image for the first time, let me know if it works for you too! 🙂

Copy link
Member

@julian-risch julian-risch left a comment

Choose a reason for hiding this comment

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

Ready to merge. 👍 I can confirm that the log messages about nltk are gone now and the caching works for me as well.

@ZanSara ZanSara merged commit 9c3d9b4 into master Jan 11, 2022
@ZanSara ZanSara deleted the cache_models_in_docker branch January 11, 2022 15:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
topic:docker type:feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add model to demo docker image
2 participants