diff --git a/examples/convert/convert_hf_to_st.py b/examples/convert/convert_hf_to_st.py index 876e06893..8af4fb466 100644 --- a/examples/convert/convert_hf_to_st.py +++ b/examples/convert/convert_hf_to_st.py @@ -26,7 +26,7 @@ def main(): hf_model = Transformer(args.model_name_or_path, tokenizer_args=tokenizer_kwargs, torch_dtype=args.torch_dtype) pooler = Pooling(hf_model.get_word_embedding_dimension(), pooling_mode=args.pooling_mode) - model = SentenceTransformer(modules=[hf_model, pooler], device='cpu') + model = SentenceTransformer(modules=[hf_model, pooler], device="cpu") model.save(args.save_path)