Skip to content

Commit 0e68f5b

Browse files
Fix get_pipeline_selection_option to disable properly (#2586)
1 parent 86dacc4 commit 0e68f5b

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

notebooks/catvton/ov_catvton_helper.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -233,13 +233,12 @@ def get_compiled_automasker(automasker, core, device, densepose_processor_path,
233233
return automasker
234234

235235

236-
def get_pipeline_selection_option(optimized_pipe=None):
236+
def get_pipeline_selection_option(is_optimized_pipe_available=False):
237237
import ipywidgets as widgets
238238

239-
model_available = optimized_pipe is not None
240239
use_quantized_models = widgets.Checkbox(
241-
value=model_available,
240+
value=is_optimized_pipe_available,
242241
description="Use quantized models",
243-
disabled=not model_available,
242+
disabled=not is_optimized_pipe_available,
244243
)
245244
return use_quantized_models

0 commit comments

Comments
 (0)