fix unit trest

Author: xadupre

_unittests/ut_torch_models/test_hghub_model.py

@@ -78,7 +78,7 @@ def test_get_untrained_model_with_inputs_beit(self):
         model, inputs = data["model"], data["inputs"]
         model(**inputs)
         # different expected value for different version of transformers
-        self.assertIn((data["size"], data["n_weights"]), [(111448, 27862)])
+        self.assertIn((data["size"], data["n_weights"]), [(111448, 27862), (56880, 14220)])
 
     @hide_stdout()
     def test_get_untrained_model_with_inputs_codellama(self):