Fix Lfm2 and common tests (#39398)

* fix

* better fix

* typo
This commit is contained in:
Cyril Vallez
2025-07-14 12:02:59 +02:00
committed by GitHub
parent 878d60a3cb
commit 8165c703ab
2 changed files with 7 additions and 0 deletions

View File

@@ -4484,6 +4484,7 @@ class ModelTesterMixin:
),
"position_ids": torch.arange(0, 10, device=torch_device).unsqueeze(0),
"labels": torch.randint(low=1, high=model.config.vocab_size, size=(2, 10), device=torch_device),
"use_cache": False,
}
# eager backward