SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Compile] Only test compiling model forward pass (#35658)

Browse Source 
* rename test to only compile forward!

* style emu
This commit is contained in:
Arthur
2025-01-13 13:43:29 +01:00
committed by GitHub
parent 84a6789145
commit e6f9b03464
7 changed files with 9 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/emu3/test_modeling_emu3.py
View File

@@ -176,10 +176,6 @@ class Emu3Text2TextModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTe
def test_custom_4d_attention_mask(self):
pass
@unittest.skip("Fails with unknown error only on end-to-end compile") # TODO raushan fixme
def test_generate_compile_1_end_to_end(self):
pass
class Emu3Vision2TextModelTester:
def __init__(
@@ -398,10 +394,6 @@ class Emu3Vision2TextModelTest(ModelTesterMixin, GenerationTesterMixin, Pipeline
def test_initialization(self):
pass
@unittest.skip("End-to-end compilation is not supported due to dynamic control in `prepare_inputs_for_generation`")
def test_generate_compile_1_end_to_end(self):
pass
@require_torch
class Emu3IntegrationTest(unittest.TestCase):