mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 12:50:06 +06:00
fix dots1
tests (#39088)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
parent
c8764ab935
commit
6d773fc3bc
@ -87,6 +87,10 @@ class Dots1ModelTest(CausalLMModelTest, unittest.TestCase):
|
||||
test_pruning = False
|
||||
model_tester_class = Dots1ModelTester
|
||||
|
||||
@unittest.skip("dots.llm1's moe is not compatible `token_indices, weight_indices = torch.where(mask)`.")
|
||||
def test_generate_with_static_cache(self):
|
||||
pass
|
||||
|
||||
@unittest.skip("dots.llm1's moe is not compatible `token_indices, weight_indices = torch.where(mask)`.")
|
||||
def test_generate_compilation_all_outputs(self):
|
||||
pass
|
||||
|
Loading…
Reference in New Issue
Block a user