update

huggingface · Oct 31, 2024 · 35b99b3 · 35b99b3
1 parent 4403c5a
commit 35b99b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/mistral/test_modeling_mistral.py b/tests/models/mistral/test_modeling_mistral.py
@@ -23,6 +23,7 @@
 
 from transformers import AutoTokenizer, MistralConfig, is_torch_available, set_seed
 from transformers.testing_utils import (
+    backend_empty_cache,
     cleanup,
     require_bitsandbytes,
     require_flash_attn,