mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-03-16 11:47:04 +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)
|
noise_uncond, noise = noise_pred_fn(x_in, t_in, cond=c_in).chunk(2)
|
||||||
return noise_uncond + guidance_scale * (noise - noise_uncond)
|
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"]
|
assert guidance_type in ["uncond", "classifier", "classifier-free"]
|
||||||
return model_fn
|
return model_fn
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user