"vscode:/vscode.git/clone" did not exist on "ad2a1c7acca3e1f70f1691da9fe139278b11c75d"
  • Yasyf Mohamedali's avatar
    Support training SD V2 with Flax (#1783) · 856331c6
    Yasyf Mohamedali authored
    * Support training SD V2 with Flax
    
    Mostly involves supporting a v_prediction scheduler.
    
    The implementation in #1777 doesn't take into account a recent refactor of `scheduling_utils_flax`, so this should be used instead.
    
    * Add to other top-level files.
    856331c6
textual_inversion_flax.py 24.7 KB