Skip to content

X0 pred#1

Draft
adsfssd wants to merge 10 commits intomainfrom
x0pred
Draft

X0 pred#1
adsfssd wants to merge 10 commits intomainfrom
x0pred

Conversation

@adsfssd
Copy link
Owner

@adsfssd adsfssd commented Jan 11, 2026

https://arxiv.org/pdf/2511.13720

adds:

  --jit_loss_type {v,x0,eps}
                        Loss space to use according to JIT
  --jit_pred_type {v,x0,eps}
                        Model prediction space to use according to JIT
  --jit_loss_weights x0_mult v_mult eps_mult
                        Perform weighted average across loss types
  --jit_t_eps JIT_T_EPS
                        t_eps to avoid division errors when converting loss types, ignored if implicit scale is used
  --jit_implicit_scale  Calculate loss in a scaled space where possible to avoid division instability

to sdxl_train.py

@adsfssd adsfssd marked this pull request as draft January 11, 2026 12:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant