Add model index config for transformer to make format uniform
Hi @cai-qi ! I'm Aryan from the diffusers team. Thanks for releasing such a powerful model to the open source community!
This PR is to add a config mapping to allow certain diffusers utilities and custom functionality to fetch the class from the model_index.json file.
For example, I'm trying to add training support in finetrainers (https://github.com/a-r-r-o-w/finetrainers) with this PR. It requires resolving the model class for configuration attributes here. The changes in this PR make that possible.
Similarly, there are other utilities and methods that may rely on such an attribute existing. LMK if this makes sense 🤗
Hi @a-r-r-o-w , thanks for the PR! This looks helpful and makes sense. Merging this PR.
I have added this attribute for both Dev and Fast model.