This commit is contained in:
Yingda Chen
2024-11-25 12:17:48 +08:00
parent 27bf8fab1e
commit 8b879953ef

View File

@@ -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'
])