fix tests

Author: xadupre

_unittests/ut_torch_models/test_llm_phi2.py

@@ -14,7 +14,7 @@
 
 class TestLlmPhi(ExtTestCase):
     def test_get_phi2(self):
-        data = get_phi2(num_hidden_layers=2)
+        data = get_phi2(num_hidden_layers=2, batch_size=2)
         model, inputs = data["model"], data["inputs"]
         self.assertIn("DynamicCache", string_type(inputs))
         model(**inputs)

_unittests/ut_torch_models/test_tiny_llms_bypassed.py

@@ -56,7 +56,7 @@ def debug():
 
     @ignore_warnings(UserWarning)
     def test_export_phi2_2_bypassed(self):
-        data = get_phi2(num_hidden_layers=2)
+        data = get_phi2(num_hidden_layers=2, batch_size=2)
         model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]
         self.assertEqual(
             {"attention_mask", "past_key_values", "input_ids", "position_ids"}, set(inputs)