From 067fdfdc7fcbb996a1bf6ab5bbbc11f8e3391b10 Mon Sep 17 00:00:00 2001 From: Katsumata420 Date: Thu, 29 Feb 2024 18:49:57 +0900 Subject: [PATCH] Fix format --- examples/convert/convert_hf_to_st.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)