add _repeated_blocks for ErnieImageTransformer2DModel#13496
add _repeated_blocks for ErnieImageTransformer2DModel#13496dg845 merged 4 commits intohuggingface:mainfrom
Conversation
Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
dg845
left a comment
There was a problem hiding this comment.
Thanks for the PR! I think we can remove the pytest.mark.skip decorators from the skipped tests in TestErnieImageTransformerCompile since they all pass after this fix.
|
@dg845 Thx for your review, but there is a comment: "The repeated block in this model is ErnieImageSharedAdaLNBlock. As a consequence of this, the inputs recorded for the block would vary during compilation and full compilation with fullgraph=True would trigger recompilation." in L119. Hence I am not sure if it is suitable to remove the |
|
Merging this PR as I think we can handle the other skipped |
This PR fixes failed test case:
tests/models/transformers/test_models_transformer_ernie_image.py::TestErnieImageTransformerCompile::test_torch_compile_repeated_blocks@yiyixuxu pls help review, thx!