From 96f0af41708d182e7a5ae66fe6f7f5fea4b82397 Mon Sep 17 00:00:00 2001 From: Robert Yang <141875536+ydm-amazon@users.noreply.github.com> Date: Mon, 12 Feb 2024 09:43:11 -0800 Subject: [PATCH] fix formatting --- .../python/setup/djl_python/properties_manager/hf_properties.py | 1 + 1 file changed, 1 insertion(+) diff --git a/engines/python/setup/djl_python/properties_manager/hf_properties.py b/engines/python/setup/djl_python/properties_manager/hf_properties.py index 7c21b204c3..a96caf8dcc 100644 --- a/engines/python/setup/djl_python/properties_manager/hf_properties.py +++ b/engines/python/setup/djl_python/properties_manager/hf_properties.py @@ -19,6 +19,7 @@ class HFQuantizeMethods(str, Enum): bitsandbytes4 = 'bitsandbytes4' bitsandbytes8 = 'bitsandbytes8' + def get_torch_dtype_from_str(dtype: str): if dtype == "auto": return dtype