298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-03 10:02:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 5d2475ef8052bb76747954461c2e01495db8dcd5 into d8e5662822168101afb5e08a8ba75b6eefff6e02

Browse Source
This commit is contained in:
Hongbo 2025-05-18 11:16:48 +02:00 committed by GitHub
commit 24bfa2b2b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/extra_samplers/uni_pc.py
View File

@ -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