diffusers
15782fd5 - [Pipeline utils] feat: implement push_to_hub for standalone models, schedulers as well as pipelines (#4128)

Comment changes are shownComment changes are hidden
Commit
1 year ago
[Pipeline utils] feat: implement push_to_hub for standalone models, schedulers as well as pipelines (#4128) * feat: implement push_to_hub for standalone models. * address PR feedback. * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * remove max_shard_size. * add: support for scheduler push_to_hub * enable push_to_hub support for flax schedulers. * enable push_to_hub for pipelines. * Apply suggestions from code review Co-authored-by: Lucain <lucainp@gmail.com> * reflect pr feedback. * address another round of deedback. * better handling of kwargs. * add: tests * Apply suggestions from code review Co-authored-by: Lucain <lucainp@gmail.com> * setting hub staging to False for now. * incorporate staging test as a separate job. Co-authored-by: ydshieh <2521628+ydshieh@users.noreply.github.com> * fix: tokenizer loading. * fix: json dumping. * move is_staging_test to a better location. * better treatment to tokens. * define repo_id to better handle concurrency * style * explicitly set token * Empty-Commit * move SUER, TOKEN to test * collate org_repo_id * delete repo --------- Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Lucain <lucainp@gmail.com> Co-authored-by: ydshieh <2521628+ydshieh@users.noreply.github.com>
Author
Parents
  • .github/workflows
    • File
      pr_tests.yml
  • docs/source/en/api/models
    • File
      overview.md
  • src/diffusers
    • File
      configuration_utils.py
    • models
      • File
        modeling_flax_utils.py
      • File
        modeling_utils.py
    • pipelines
      • File
        pipeline_flax_utils.py
      • File
        pipeline_utils.py
    • schedulers
      • File
        scheduling_utils.py
      • File
        scheduling_utils_flax.py
    • utils
      • File
        __init__.py
      • File
        hub_utils.py
  • tests
    • models
      • File
        test_modeling_common.py
    • others
      • File
        test_utils.py
    • pipelines
      • File
        test_pipelines_common.py
    • schedulers
      • File
        test_schedulers.py