diff --git a/modelscope/utils/hf_util.py b/modelscope/utils/hf_util.py index 57b0d3fe..7d684a44 100644 --- a/modelscope/utils/hf_util.py +++ b/modelscope/utils/hf_util.py @@ -364,8 +364,8 @@ BitsAndBytesConfig = get_wrapped_class( ignore_file_pattern=[ r'\w+\.bin', r'\w+\.safetensors', r'\w+\.pth', r'\w+\.pt', r'\w+\.h5' ]) -AutoImageProcessor = AutoImageProcessorHF( - BitsAndBytesConfigHF, +AutoImageProcessor = get_wrapped_class( + AutoImageProcessorHF, ignore_file_pattern=[ r'\w+\.bin', r'\w+\.safetensors', r'\w+\.pth', r'\w+\.pt', r'\w+\.h5' ])