mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-03 05:09:42 +08:00
Include default when complaining about invalid solver types
This commit is contained in:
parent
779346882f
commit
66bbdefcc0
@ -1555,7 +1555,7 @@ def sample_er_sde(
|
|||||||
s_in = x.new_ones(x.shape[:1])
|
s_in = x.new_ones(x.shape[:1])
|
||||||
|
|
||||||
if solver_type not in {"default", "sde", "ersde"}:
|
if solver_type not in {"default", "sde", "ersde"}:
|
||||||
raise ValueError("Bad solver_type, must be one of ersde or sde")
|
raise ValueError("Bad solver_type, must be one of default, ersde or sde")
|
||||||
if noise_scaler is None:
|
if noise_scaler is None:
|
||||||
if solver_type == "sde":
|
if solver_type == "sde":
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user