From f9742143538cfeb65c3a25632b83276266819c41 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 18 Apr 2025 10:09:19 +0200 Subject: [PATCH] Fix some GPU OOM after #37553 (#37591) * fix * trigger CI --------- Co-authored-by: ydshieh --- tests/models/llama/test_modeling_llama.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/llama/test_modeling_llama.py b/tests/models/llama/test_modeling_llama.py index fe843938cf..c9b86c128c 100644 --- a/tests/models/llama/test_modeling_llama.py +++ b/tests/models/llama/test_modeling_llama.py @@ -699,6 +699,7 @@ class Mask4DTestHard(unittest.TestCase): cleanup(torch_device, gc_collect=True) def setUp(self): + cleanup(torch_device, gc_collect=True) model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0" self.model_dtype = torch.float32 self.tokenizer = LlamaTokenizer.from_pretrained(model_name)