Support segformer fx (#19924)

* Support segformer fx

* Add fx_compatible attribute to test_modeling_segformer.py

* Update glpn model (fx support)

glpn model was copied from segformer.

* Update utils/fx.py | add semantic-segmentation

for SegformerForSemanticSegmentation model

* Fix minor import order(isort)

* Add random input generation for segformer fx

Co-authored-by: noelbird <lduldu00228@gmail.com>
This commit is contained in:
donguk.lim
2022-10-28 21:44:38 +09:00
committed by GitHub
parent dcca71be61
commit 347ba38cb4
4 changed files with 9 additions and 4 deletions

View File

@@ -161,6 +161,7 @@ class SegformerModelTest(ModelTesterMixin, unittest.TestCase):
else ()
)
fx_compatible = True
test_head_masking = False
test_pruning = False
test_resize_embeddings = False