Update convert.py

convert.py

@@ -176,6 +176,8 @@ def check_final_model(model_id: str, folder: str):
     input_ids = torch.arange(10).unsqueeze(0)
     pixel_values = torch.randn(1, 3, 224, 224)
     input_values = torch.arange(1000).float().unsqueeze(0)
+    task_inputs = torch.randn(1,77)
+    
     kwargs = {}
     if "input_ids" in sig.parameters:
         kwargs["input_ids"] = input_ids
@@ -189,6 +191,8 @@ def check_final_model(model_id: str, folder: str):
         kwargs["bbox"] = torch.zeros((1, 10, 4)).long()
     if "image" in sig.parameters:
         kwargs["image"] = pixel_values
+    if "task_inputs" in sig.parameters:
+        kwargs["task_inputs"] = task_inputs
 
     if torch.cuda.is_available():
         pt_model = pt_model.cuda()