remove redundant codes (#1762)

openvinotoolkit · Feb 28, 2024 · f66c34d · f66c34d
1 parent 722bc1a
commit f66c34d
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/notebooks/254-llm-chatbot/ov_llm_model.py b/notebooks/254-llm-chatbot/ov_llm_model.py
@@ -301,14 +301,7 @@ def __init__(
             model, config, device, dynamic_shapes, ov_config, model_save_dir, **kwargs
         )
 
-    def _reshape(
-        self,
-        model: "Model",
-        batch_size: int,
-        sequence_length: int,
-        height: int = None,
-        width: int = None,
-    ):
+    def _reshape(self, model: "Model", *args, **kwargs):
         shapes = {}
         for inputs in model.inputs:
             shapes[inputs] = inputs.get_partial_shape()
@@ -368,4 +361,4 @@ def _from_pretrained(
     "baichuan2": OVBAICHUANModel,
     "chatglm3": OVCHATGLMModel,
     "gemma": OVModelForCausalLM
-}
+}