version

xadupre · xadupre · commit 60f4304b432b · 2025-03-28T00:11:59.000+01:00
diff --git a/_unittests/ut_torch_models/test_hghub_model.py b/_unittests/ut_torch_models/test_hghub_model.py
@@ -69,7 +69,7 @@ def test_get_untrained_model_with_inputs_phi_2(self):
         # different expected value for different version of transformers
         self.assertIn(
             (data["size"], data["n_weights"]),
-            [(453330944, 113332736)],
+            [(453330944, 113332736), (453126144, 113281536)],
         )
 
     @hide_stdout()

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def test_get_untrained_model_with_inputs_phi_2(self):`
`69`	`69`	`# different expected value for different version of transformers`
`70`	`70`	`self.assertIn(`
`71`	`71`	`(data["size"], data["n_weights"]),`
`72`		`- [(453330944, 113332736)],`
	`72`	`+ [(453330944, 113332736), (453126144, 113281536)],`
`73`	`73`	`)`
`74`	`74`
`75`	`75`	`@hide_stdout()`