diffusers/examples/research_projects/diffusion_orpo/requirements.txt at 4c5082ca685d1dd0a24907890da170e258e5cd01 - diffusers - Gitea: Git with a cup of tea

obscura/diffusers

Files

T

Sayak Paul 1b917981da [Research Projects] ORPO diffusion for alignment (#7423 )

* barebones orpo

* remove reference model.

* full implementation

* change default of beta_orpo

* add a training command.

* fix: dataloading issues.

* interpreting the formulation.

* revert styling

* add: wds full blown version

* fix: per_gpu_batch_siz

* start debuggin

* debugging

* remove print

* fix

* remove filter keys.

* turn on non-blocking calls.

* device_placement

* let's see.

* add bigger training run command

* reinitialize generator for fair repro

* add: detailed readme and requirements

---------

Co-authored-by: Sayak Paul <sayakpaul@Sayaks-MacBook-Pro-2.local>

2024-12-23 13:02:09 +05:30

7 lines

66 B

Plaintext

Raw Blame History

 datasets
 accelerate
 transformers
 torchvision
 wandb
 peft
 webdataset