Browse Source

Add `"score"` as a supported `model_type`

Originally, `"score"` is not an allowed model type. However, the documentation supports `"score"` and the method has code related to it. This commit adds it back.
pull/2918/head
Hongbo 9 months ago committed by GitHub
parent
commit
5d2475ef80
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      comfy/extra_samplers/uni_pc.py

4
comfy/extra_samplers/uni_pc.py

@ -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
@ -872,4 +872,4 @@ def sample_unipc(model, noise, sigmas, extra_args=None, callback=None, disable=F
return x
def sample_unipc_bh2(model, noise, sigmas, extra_args=None, callback=None, disable=False):
return sample_unipc(model, noise, sigmas, extra_args, callback, disable, variant='bh2')
return sample_unipc(model, noise, sigmas, extra_args, callback, disable, variant='bh2')

Loading…
Cancel
Save