mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-03-16 10:17:06 +08:00
Merge 5d2475ef8052bb76747954461c2e01495db8dcd5 into 33aa808713f7c36cd9476c53b8b67c745e9bc107
This commit is contained in:
commit
6168dc05f9
@ -344,7 +344,7 @@ def model_wrapper(
|
||||
noise_uncond, noise = noise_pred_fn(x_in, t_in, cond=c_in).chunk(2)
|
||||
return noise_uncond + guidance_scale * (noise - noise_uncond)
|
||||
|
||||
assert model_type in ["noise", "x_start", "v"]
|
||||
assert model_type in ["noise", "x_start", "v", "score"]
|
||||
assert guidance_type in ["uncond", "classifier", "classifier-free"]
|
||||
return model_fn
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user