gmastrapas commited on
Commit
8986313
1 Parent(s): e7432eb

fix: kwargs in custom Sentence Transformer

Browse files
Files changed (1) hide show
  1. 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