Skip to content

Commit f0a6ead

Browse files
committed
enable fake tensor more
1 parent da9cb89 commit f0a6ead

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

_unittests/ut_torch_models/test_validate_whole_models1.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,21 @@ def test_n_validate_phi35_mini_instruct(self):
225225
self.assertIn("If", op_types)
226226
self.clean_dump()
227227

228+
@hide_stdout()
229+
def test_o_validate_model_export_fake(self):
230+
mid = "arnir0/Tiny-LLM"
231+
summary, data = validate_model(
232+
mid,
233+
do_run=True,
234+
verbose=10,
235+
exporter="custom-fake",
236+
dump_folder="dump_test/validate_model_export_fake",
237+
patch=True,
238+
)
239+
self.assertIsInstance(summary, dict)
240+
self.assertIsInstance(data, dict)
241+
self.clean_dump()
242+
228243

229244
if __name__ == "__main__":
230245
unittest.main(verbosity=2)

0 commit comments

Comments
 (0)