Better component registry and import structure - Githubissues

Modalities / modalities

A framework for training multimodal foundation models.

MIT License

39 stars 3 forks source link

Better component registry and import structure #90

Open spravil opened 4 months ago

spravil commented 4 months ago

Use a decorator function to register new components. Import all components of a submodule in the __init__.py to simplify imports (from modalities.models import CoCa, CoCaConfig)

This should address the following points:

[ ] Move config classes closer to their corresponding class (GPT2LLMCollateFnConfig to modalities.models.gpt2.collator)
[ ] Decorator function to register new components
[ ] Collect all components of a submodule in the __init__.py.