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
35 stars 6 forks source link

[Feature Request] - Allow using an ENV_VAR or something to set "checkpoints" path. #36

Closed d8ahazard closed 1 month ago

d8ahazard commented 1 month ago

Is your feature request related to a problem? Please describe. Right now, there is no apparent way to set a custom location to save/load the checkpoints to. It would be ideal to be able to set an environment variable so the library knows what path to look in for models.

Describe the solution you'd like Set a "VID_GEN_HUB" environment variable with the path.

Describe alternatives you've considered A config file, a param I can pass to the model loader, etc.

Additional context Add any other context or screenshots about the feature request here.