gmastrapas
commited on
Commit
•
8986313
1
Parent(s):
e7432eb
fix: kwargs in custom Sentence Transformer
Browse files- custom_st.py +1 -1
custom_st.py
CHANGED
@@ -98,7 +98,7 @@ class Transformer(nn.Module):
|
|
98 |
'revision': model_kwargs.get('revision', None),
|
99 |
'local_files_only': model_kwargs.get('local_files_only', None),
|
100 |
}
|
101 |
-
image_processor_kwargs.update(image_processor_args)
|
102 |
|
103 |
config = AutoConfig.from_pretrained(
|
104 |
model_name_or_path, cache_dir=cache_dir, **config_kwargs
|
|
|
98 |
'revision': model_kwargs.get('revision', None),
|
99 |
'local_files_only': model_kwargs.get('local_files_only', None),
|
100 |
}
|
101 |
+
image_processor_kwargs.update(image_processor_args or {})
|
102 |
|
103 |
config = AutoConfig.from_pretrained(
|
104 |
model_name_or_path, cache_dir=cache_dir, **config_kwargs
|