Commit 62f90a8
committed
Fix AttributeError in diffusion_dpo ipo loss path
The IPO branch in `train_diffusion_dpo.py` references `args.beta`, but
this script only defines `--beta_dpo` — there is no `--beta` argument.
Running training with `--loss_type ipo` crashes immediately at the loss
computation with:
AttributeError: 'Namespace' object has no attribute 'beta'
The `sigmoid` and `hinge` branches right above correctly use
`args.beta_dpo`, so IPO was clearly meant to use the same hyperparameter.
Swap to `args.beta_dpo` to unbreak `--loss_type ipo`.1 parent c8c8401 commit 62f90a8
1 file changed
Lines changed: 1 addition & 1 deletion
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
881 | 881 | | |
882 | 882 | | |
883 | 883 | | |
884 | | - | |
| 884 | + | |
885 | 885 | | |
886 | 886 | | |
887 | 887 | | |
| |||
0 commit comments