search

TIGER-AI-Lab / VideoGenHub

A one-stop library to standardize the inference and evaluation of all the conditional video generation models.
https://pypi.org/project/videogen-hub/
MIT License
39 stars 7 forks source link

Add universal model path via ENV_VAR, fix other issues on the way #37

Closed d8ahazard closed 3 months ago

d8ahazard commented 3 months ago

If the VIDEO_MODEL_PATH env var is set, the app will use that path, instead of ./checkpoints.

This should address #36

I also added a missing req. for SEINE, as well as fix an import conflict with the "datasets" library.

d8ahazard commented 3 months ago

OK, sorry, I honestly didn't mean to just start fixing "all the things", but I'm working on Windows (just to test the model stuff), and after looking at the code, I realized it probably doesn't have to be linux-only.

vinesmsuic commented 3 months ago

Hi, really appreciate the effort for fixing the issues! I am now reviewing the changes and will try to get it merged.

d8ahazard commented 3 months ago

Hi, really appreciate the effort for fixing the issues! I am now reviewing the changes and will try to get it merged.

I may still have some fixing to do. Trying to use an exported wheel of this, I'm getting an issue importing MODEL_PATH. yet. Might need to make it a separate file.

Or, it might just be me exporting it to a different package name. Just be sure to test that the new package before publishing to pypi. ;)

vinesmsuic commented 3 months ago

Just got merged. Thanks for the code improvement! We will extend the codebase in the future, following a similar ecosystem in ImagenHub.