geniverse-hub 0.0.0

Library for downloading and loading generative AI models.

geniverse_hub 🪐

Use and contribute to the Geniverse HUB 🪐 with the following instructions.

Setup

First install the geniverse_hub python package with the following command:

pip install geniverse_hub

This will provide you with the required scripts to download and utilize the models available in the hub.

All the models are extracted from github repositories. Hence, one requirement for this library to work is to have Git installed and working in your machine. All the available models are referenced in this config file with its owner, name and ditribution (i.e., the branch to be used within the github repo).

Usage

There are some common functionalities for downloading and loading models in the hub. The following is an example on how to use VQGAN:

from geniverse_hub import hub_utils
taming_utils = hub_utils.load_from_hub(module_name="taming", )
model = taming_utils.load_model()

You can utilize any of the models in the hub's config file by changing the value of module_name for the corresponding model that you intend to load.

The result will be equivelant to a python package where you will have access to functionalities of the specific module that you loaded such as loading a model. In order to find the functionalities for each of the models, access their respective documentation in https://github.com/<owner>/<name>/tree/<dist> where owner, name and dist can be found in the hub's config file.

Contribute

In order to contribute to the Geniverse Hub 🪐 you just need to create a github repository following some simple guidelines and reference it within our config file.

Contribution Guidelines

In order for your model to be accessible from the hub, you must provide with a requirements.txt file where all the python dependencies are stated. The repository should contain a python package with the same name os the repository. This package should contain a modeling_utils.py script where the functionalities for your model are available. The result from geniverse_hub.hub_utils.load_module(<your-module's-name>) will be the set of functions and classes that you implement there.

You can find an example of this coding structure in our taming implementation in https://github.com/thegeniverse/taming/tree/master. Here, we have a python package with the same name as the repo (taming) with all the required code for this module to work. There, we can also find the modeling_utils.py. Inside this file there are all the functionalities that we give access for anyone using our model from the hub. For example, we can find the function load_model() that can be used to load a VQGAN model.