From 5bd29983e0f12a68b8afaa48c63229622b96ff59 Mon Sep 17 00:00:00 2001 From: Sai-Suraj-27 Date: Tue, 16 Jul 2024 15:10:21 +0530 Subject: [PATCH] Fixed incorrect dictionary assignment. --- src/transformers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 11d442a4e2808a..344da0d3dc8ef3 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -1286,7 +1286,7 @@ "WhisperTimeStampLogitsProcessor", ] ) - _import_structure["modeling_flash_attention_utils"]: [] + _import_structure["modeling_flash_attention_utils"] = [] _import_structure["modeling_outputs"] = [] _import_structure["modeling_utils"] = ["PreTrainedModel"]