Age | Commit message (Collapse) | Author |
|
Implement zero terminal SNR noise schedule option
|
|
|
|
|
|
|
|
|
|
|
|
on load
|
|
|
|
|
|
|
|
|
|
|
|
and add a readme entry
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Since norm layer need fp32, I only convert the linear operation layer(conv2d/linear)
And TE have some pytorch function not support bf16 amp in CPU. I add a condition to indicate if the autocast is for unet.
|
|
|
|
|
|
|
|
|
|
fix: checkpoints_loaded:{checkpoint:state_dict}, model.load_state_dict issue in dict value empty
|
|
|
|
|
|
fix `--ckpt-dir` path separator and option use `short name` for checkpoint dropdown
|
|
|
|
patch DDPM.register_betas so that users can put given_betas in model yaml
|
|
|
|
|
|
|
|
|
|
checksum
|
|
checksum
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
This reverts commit 0815c45bcdec0a2e5c60bdd5b33d95813d799c01.
|