Skip to content

Commit 2b4193c

Browse files
remove the 'remove compiled model'
1 parent 6c58973 commit 2b4193c

2 files changed

Lines changed: 2 additions & 9 deletions

File tree

node_openvino.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -151,15 +151,13 @@ def define_schema(cls) -> io.Schema:
151151
io.Combo.Input("device", options=available_devices),
152152
io.Boolean.Input("compile_encoder", default=True),
153153
io.Boolean.Input("compile_decoder", default=True),
154-
io.Boolean.Input("remove_compile", default=False,
155-
tooltip="Remove VAE compilation"),
156154
],
157155
outputs=[io.Vae.Output()],
158156
is_experimental=True,
159157
)
160158

161159
@classmethod
162-
def execute(cls, vae, device, compile_encoder, compile_decoder, remove_compile) -> io.NodeOutput:
160+
def execute(cls, vae, device, compile_encoder, compile_decoder) -> io.NodeOutput:
163161
torch._dynamo.reset()
164162
ov_ex.compiled_cache.clear()
165163
ov_ex.req_cache.clear()
@@ -171,11 +169,6 @@ def execute(cls, vae, device, compile_encoder, compile_decoder, remove_compile)
171169

172170
wrapper = vae._compile_wrapper
173171

174-
# Remove compilation if requested
175-
if remove_compile:
176-
wrapper.remove()
177-
return io.NodeOutput(vae)
178-
179172
# Otherwise compile as requested
180173
keys = []
181174
first_stage = vae.first_stage_model

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[project]
22
name = "comfyui-openvino"
33
description = "OpenVINO node is designed for optimizing the performance of model inference in ComfyUI by leveraging Intel OpenVINO toolkits. It can support running model on Intel CPU, GPU and NPU device."
4-
version = "1.1.3"
4+
version = "1.1.4"
55
license = {file = "LICENSE"}
66

77
[project.urls]

0 commit comments

Comments
 (0)