Skip to content

Commit 6851268

Browse files
guangy10facebook-github-bot
authored andcommitted
Remove emformer model
Summary: As titled, because this model is not fully ready yet. Reviewed By: kirklandsign Differential Revision: D48334229 fbshipit-source-id: 6429a8161723e3570b47f050bb0f513a395d356b
1 parent 889e463 commit 6851268

File tree

6 files changed

+0
-86
lines changed

6 files changed

+0
-86
lines changed

examples/export/test/test_export.py

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -74,23 +74,6 @@ def test_mv2_export_to_executorch(self):
7474
eager_model, example_inputs, self.validate_tensor_allclose
7575
)
7676

77-
def test_emformer_export_to_executorch(self):
78-
eager_model, example_inputs = MODEL_NAME_TO_MODEL["emformer"]()
79-
eager_model = eager_model.eval()
80-
81-
validate_emformer_result = (
82-
lambda eager_output, executorch_output: torch.allclose(
83-
eager_output[0],
84-
executorch_output[0][0],
85-
rtol=1e-5,
86-
atol=1e-5,
87-
)
88-
)
89-
90-
self._assert_eager_lowered_same_result(
91-
eager_model, example_inputs, validate_emformer_result
92-
)
93-
9477
def test_vit_export_to_executorch(self):
9578
eager_model, example_inputs = MODEL_NAME_TO_MODEL["vit"]()
9679
eager_model = eager_model.eval()

examples/models/TARGETS

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ python_library(
88
],
99
deps = [
1010
"//caffe2:torch",
11-
"//executorch/examples/models/emformer:emformer_export",
1211
"//executorch/examples/models/mobilenet_v2:mv2_export",
1312
"//executorch/examples/models/mobilenet_v3:mv3_export",
1413
"//executorch/examples/models/torchvision_vit:vit_export",

examples/models/emformer/TARGETS

Lines changed: 0 additions & 14 deletions
This file was deleted.

examples/models/emformer/__init__.py

Lines changed: 0 additions & 11 deletions
This file was deleted.

examples/models/emformer/export.py

Lines changed: 0 additions & 36 deletions
This file was deleted.

examples/models/models.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -89,12 +89,6 @@ def gen_mobilenet_v2_model_inputs() -> Tuple[torch.nn.Module, Any]:
8989
return MV2Model.get_model(), MV2Model.get_example_inputs()
9090

9191

92-
def gen_emformer_model_inputs() -> Tuple[torch.nn.Module, Any]:
93-
from ..models.emformer import EmformerModel
94-
95-
return EmformerModel.get_model(), EmformerModel.get_example_inputs()
96-
97-
9892
def gen_torchvision_vit_model_and_inputs() -> Tuple[torch.nn.Module, Any]:
9993
from ..models.torchvision_vit import TorchVisionViTModel
10094

@@ -108,6 +102,5 @@ def gen_torchvision_vit_model_and_inputs() -> Tuple[torch.nn.Module, Any]:
108102
"add_mul": lambda: (AddMulModule(), AddMulModule.get_example_inputs()),
109103
"mv2": gen_mobilenet_v2_model_inputs,
110104
"mv3": gen_mobilenet_v3_model_inputs,
111-
"emformer": gen_emformer_model_inputs,
112105
"vit": gen_torchvision_vit_model_and_inputs,
113106
}

0 commit comments

Comments
 (0)