Some test in test_hooked_model need to be run two times to pass #16
Description
It could be due to some initialization configuration. An example is the text_extract_cache_with_gradient_computations.