Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Suraj Patil's avatar
    v-prediction training support (#1455) · 6c56f050
    Suraj Patil authored 
    * add get_velocity

* add v prediction for training

* fix saving

* add revision arg

* fix saving

* save checkpoints dreambooth

* fix saving embeds

* add instruction in readme

* quality

* noise_pred -> model_pred
    6c56f050
scheduling_ddpm.py 16.6 KB