From d9e9e4af9e6f98d1a16b308acbb7a812f9737e8a Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Fri, 15 Nov 2024 15:24:11 +0100 Subject: [PATCH] fix --- optimum/utils/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/optimum/utils/__init__.py b/optimum/utils/__init__.py index db7d1f6975d..9afcdcb9117 100644 --- a/optimum/utils/__init__.py +++ b/optimum/utils/__init__.py @@ -33,6 +33,7 @@ check_if_transformers_greater, is_accelerate_available, is_auto_gptq_available, + is_datasets_available, is_diffusers_available, is_onnx_available, is_onnxruntime_available,