Skip to content

Commit 0ca04f6

Browse files
committed
delete maybe_free_model_hooks from pipeline_visualcloze_combined
1 parent 26750b1 commit 0ca04f6

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/diffusers/pipelines/visualcloze/pipeline_visualcloze_combined.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -380,9 +380,6 @@ def __call__(
380380
output_type=output_type if upsampling_strength == 0 else "pil",
381381
)
382382
if upsampling_strength == 0:
383-
# Offload all models
384-
self.maybe_free_model_hooks()
385-
386383
if not return_dict:
387384
return (generation_output,)
388385

@@ -441,9 +438,6 @@ def __call__(
441438
else:
442439
output = image
443440

444-
# Offload all models
445-
self.maybe_free_model_hooks()
446-
447441
if not return_dict:
448442
return (output,)
449443

0 commit comments

Comments
 (0)