mindspore-lab / mindone

one for all, Optimal generator with No Exception
https://mindspore-lab.github.io/mindone/
Apache License 2.0
353 stars 67 forks source link

feat(diffusers/utils): import&export utils #543

Closed townwish4git closed 3 months ago

townwish4git commented 3 months ago

What does this PR do?

Add some utils in mindone.diffusers, including:

1. import_utils

BACKENDS_MAPPING containing bs, opencv, ftfy to check if it is available for importing or not

2. export_utils

export utils such as export_to_gif, export_to_obj and etc to export outputs of models to specific file format

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag members/contributors who may be interested in your PR.

@geniuspatrick @CaitinZhao