Code block

Author: stephantul

README.md

@@ -112,7 +112,10 @@ vocabulary = ["word1", "word2", "word3"]
 model_name = "BAAI/bge-base-en-v1.5"
 
 # Distill the model with the custom vocabulary
-m2v_model = distill(model_name=model_name, vocabulary=vocabulary, pca_dims=None, apply_zipf=True)
+m2v_model = distill(model_name=model_name,
+                    vocabulary=vocabulary,
+                    pca_dims=None,
+                    apply_zipf=True)
 
 # Save the model
 m2v_model.save_pretrained("m2v_model")