Skip to content

Commit 891081d

Browse files
revert to 1.1.3
1 parent 330fec9 commit 891081d

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

node_openvino.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,13 +151,15 @@ 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"),
154156
],
155157
outputs=[io.Vae.Output()],
156158
is_experimental=True,
157159
)
158160

159161
@classmethod
160-
def execute(cls, vae, device, compile_encoder, compile_decoder) -> io.NodeOutput:
162+
def execute(cls, vae, device, compile_encoder, compile_decoder, remove_compile) -> io.NodeOutput:
161163
torch._dynamo.reset()
162164
ov_ex.compiled_cache.clear()
163165
ov_ex.req_cache.clear()
@@ -169,6 +171,11 @@ def execute(cls, vae, device, compile_encoder, compile_decoder) -> io.NodeOutput
169171

170172
wrapper = vae._compile_wrapper
171173

174+
# Remove compilation if requested
175+
if remove_compile:
176+
wrapper.remove()
177+
return io.NodeOutput(vae)
178+
172179
# Otherwise compile as requested
173180
keys = []
174181
first_stage = vae.first_stage_model

0 commit comments

Comments
 (0)