Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
8d80584f
"configs/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "185d1e1f929642e4bae576c3888b9500771a84fe"
Commit
8d80584f
authored
Nov 12, 2023
by
comfyanonymous
Browse files
Remove useless argument from uni_pc sampler.
parent
248aa3e5
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
comfy/extra_samplers/uni_pc.py
comfy/extra_samplers/uni_pc.py
+1
-1
comfy/samplers.py
comfy/samplers.py
+2
-2
No files found.
comfy/extra_samplers/uni_pc.py
View file @
8d80584f
...
...
@@ -858,7 +858,7 @@ def predict_eps_sigma(model, input, sigma_in, **kwargs):
return
(
input
-
model
(
input
,
sigma_in
,
**
kwargs
))
/
sigma
def
sample_unipc
(
model
,
noise
,
image
,
sigmas
,
sampling_function
,
max_denoise
,
extra_args
=
None
,
callback
=
None
,
disable
=
False
,
noise_mask
=
None
,
variant
=
'bh1'
):
def
sample_unipc
(
model
,
noise
,
image
,
sigmas
,
max_denoise
,
extra_args
=
None
,
callback
=
None
,
disable
=
False
,
noise_mask
=
None
,
variant
=
'bh1'
):
timesteps
=
sigmas
.
clone
()
if
sigmas
[
-
1
]
==
0
:
timesteps
=
sigmas
[:]
...
...
comfy/samplers.py
View file @
8d80584f
...
...
@@ -511,11 +511,11 @@ class Sampler:
class
UNIPC
(
Sampler
):
def
sample
(
self
,
model_wrap
,
sigmas
,
extra_args
,
callback
,
noise
,
latent_image
=
None
,
denoise_mask
=
None
,
disable_pbar
=
False
):
return
uni_pc
.
sample_unipc
(
model_wrap
,
noise
,
latent_image
,
sigmas
,
sampling_function
=
sampling_function
,
max_denoise
=
self
.
max_denoise
(
model_wrap
,
sigmas
),
extra_args
=
extra_args
,
noise_mask
=
denoise_mask
,
callback
=
callback
,
disable
=
disable_pbar
)
return
uni_pc
.
sample_unipc
(
model_wrap
,
noise
,
latent_image
,
sigmas
,
max_denoise
=
self
.
max_denoise
(
model_wrap
,
sigmas
),
extra_args
=
extra_args
,
noise_mask
=
denoise_mask
,
callback
=
callback
,
disable
=
disable_pbar
)
class
UNIPCBH2
(
Sampler
):
def
sample
(
self
,
model_wrap
,
sigmas
,
extra_args
,
callback
,
noise
,
latent_image
=
None
,
denoise_mask
=
None
,
disable_pbar
=
False
):
return
uni_pc
.
sample_unipc
(
model_wrap
,
noise
,
latent_image
,
sigmas
,
sampling_function
=
sampling_function
,
max_denoise
=
self
.
max_denoise
(
model_wrap
,
sigmas
),
extra_args
=
extra_args
,
noise_mask
=
denoise_mask
,
callback
=
callback
,
variant
=
'bh2'
,
disable
=
disable_pbar
)
return
uni_pc
.
sample_unipc
(
model_wrap
,
noise
,
latent_image
,
sigmas
,
max_denoise
=
self
.
max_denoise
(
model_wrap
,
sigmas
),
extra_args
=
extra_args
,
noise_mask
=
denoise_mask
,
callback
=
callback
,
variant
=
'bh2'
,
disable
=
disable_pbar
)
KSAMPLER_NAMES
=
[
"euler"
,
"euler_ancestral"
,
"heun"
,
"dpm_2"
,
"dpm_2_ancestral"
,
"lms"
,
"dpm_fast"
,
"dpm_adaptive"
,
"dpmpp_2s_ancestral"
,
"dpmpp_sde"
,
"dpmpp_sde_gpu"
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment