Skip to content

Conversation

@Birch-san
Copy link
Owner

Note: for a cleaner diff: merge-base of this PR is set to an unrelated branch in my repository, rather than upstreaming to crowsonkb/k-diffusion. because this branch is downstream of my Mac fixes and miscellany.

I made the model wrapper generic, so that you get auto-completion from inner_model:
image

somewhat worried about this:

sigma_hat = model.discretize_sigma(sigma_hat)

since using quantize=True and churn=0.: we should expect this to be a no-op.

yet, there's a slight difference (see nose) when we apply this roundtrip to the sigma (left = usual, right = discretization):

when churn>0, the difference is, uh, more dramatic (left = discretized, churn=0; right = discretized, churn = 0.1):

churn>0 does work if I remove the discretization (left = non-discretized, churn=0; right = non-discretized, churn = 0.1):

Birch-san added 3 commits May 14, 2023 23:15
…can make k-diffusion wrapper compatible classes (e.g. which delegate to a k-diffusion wrapper rather than inheriting from one)
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.

2 participants