fix

app.py

@@ -72,14 +72,14 @@ def parse_configs(inputs):
         task=task,
         model=model,
         device=DEVICE,
-        **configs["pytorch"],
+        **{k: v for k, v in configs["pytorch"] if v},
     )
 
     configs["openvino"] = OVConfig(
         task=task,
         model=openvino_label or model,
         device=DEVICE,
-        **configs["openvino"],
+        **{k: v for k, v in configs["openvino"] if v},
     )
 
     return configs

config_store.py

@@ -78,21 +78,26 @@ def get_pytorch_config():
 
 def get_openvino_config():
     return {
+        "openvino.half": gr.Checkbox(
+            value=False,
+            label="openvino.half",
+            info="Converts model to half precision",
+        ),
         "openvino.reshape": gr.Checkbox(
             value=False,
             label="openvino.reshape",
             info="Reshapes the model to the input shape",
         ),
+        "openvino.reshape_kwargs": gr.Textbox(
+            label="openvino.reshape_kwargs",
+            value="{'batch_size': 2, 'sequence_length': 16}",
+            info="Additional python dict of kwargs to pass to the reshape function",
+        ),
         "openvino.compile": gr.Checkbox(
             value=False,
             label="openvino.compile",
             info="Compiles model for the current device",
         ),
-        "openvino.half": gr.Checkbox(
-            value=False,
-            label="openvino.half",
-            info="Converts model to half precision",
-        ),
         "openvino.load_in_8bit": gr.Checkbox(
             value=False,
             label="openvino.load_in_8bit",

pre-requirements.txt

@@ -1,7 +1,7 @@
+pip==24.3.1
 --index-url https://download.pytorch.org/whl/cpu
 torch
 --index-url https://download.pytorch.org/whl/cpu
 torchvision
 --index-url https://download.pytorch.org/whl/cpu
 torchaudio
-pip==24.3.1