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
94d5a128
Commit
94d5a128
authored
May 01, 2024
by
comfyanonymous
Browse files
Don't load the model in SDTurboScheduler
parent
2aed53c4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
comfy_extras/nodes_custom_sampler.py
comfy_extras/nodes_custom_sampler.py
+1
-2
No files found.
comfy_extras/nodes_custom_sampler.py
View file @
94d5a128
...
...
@@ -107,8 +107,7 @@ class SDTurboScheduler:
def
get_sigmas
(
self
,
model
,
steps
,
denoise
):
start_step
=
10
-
int
(
10
*
denoise
)
timesteps
=
torch
.
flip
(
torch
.
arange
(
1
,
11
)
*
100
-
1
,
(
0
,))[
start_step
:
start_step
+
steps
]
comfy
.
model_management
.
load_models_gpu
([
model
])
sigmas
=
model
.
model
.
model_sampling
.
sigma
(
timesteps
)
sigmas
=
model
.
get_model_object
(
"model_sampling"
).
sigma
(
timesteps
)
sigmas
=
torch
.
cat
([
sigmas
,
sigmas
.
new_zeros
([
1
])])
return
(
sigmas
,
)
...
...
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