making offline loading of STFR default

2024-12-20 14:32:06 +01:00
parent 80a35c4658
commit ef5743bc85
2 changed files with 6 additions and 3 deletions
--- a/tests/test_model_loader.py
+++ b/tests/test_model_loader.py
@@ -42,6 +42,7 @@ def test_load_sentence_transformer(
        backend=STFRBackends.TORCH,
        device=STFRDeviceTypes.CPU,
        model_kwargs=None,
+        local_files_only=False,
    )
    assert isinstance(model, SentenceTransformer)

@@ -70,6 +71,7 @@ def test_load_sentence_transformer_onnx(model_name, similarity_func) -> None:
        backend=STFRBackends.ONNX,
        device=STFRDeviceTypes.CPU,
        model_kwargs=stfr_model_args_onnx,  # type: ignore
+        local_files_only=False,
    )
    assert isinstance(model, SentenceTransformer)