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 for ensemble model zip files #311

Merged
merged 4 commits into from
Mar 28, 2024
Merged

Support for ensemble model zip files #311

merged 4 commits into from
Mar 28, 2024

Conversation

stefdoerr
Copy link
Collaborator

Allow also passing a single zip file containing many ckpt instead of a list of ckpt files

@stefdoerr stefdoerr requested a review from RaulPPelaez March 27, 2024 14:58
@stefdoerr stefdoerr self-assigned this Mar 27, 2024
@RaulPPelaez
Copy link
Collaborator

Load_model was getting a bit cluttered so I moved the Ensemble loading functionality to a different function. Also added some assertions and updated the docs.

I think it is fine to have the imports globally, everything is in python proper.

Feel free to add/disregard and merge yourself when you are ready.
Thanks!

@stefdoerr stefdoerr merged commit 8b47246 into main Mar 28, 2024
2 checks passed
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.

2 participants