diff --git a/tests/test_translate.py b/tests/test_translate.py index c832637..323f2e7 100644 --- a/tests/test_translate.py +++ b/tests/test_translate.py @@ -1,5 +1,5 @@ from hf_hub_ctranslate2 import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub, MultiLingualTranslatorCT2fromHfHub -from hf_hub_ctranslate2.util import _download_model +from hf_hub_ctranslate2.util import utils as _utils from transformers import AutoTokenizer @@ -51,7 +51,7 @@ def test_generator(model_name="michaelfeil/ct2fast-pythia-160m"): def test_generator_single(model_name="michaelfeil/ct2fast-pythia-160m"): - model_path = _download_model(model_name) + model_path = _utils._download_model(model_name) tokenizer = AutoTokenizer.from_pretrained(model_path) model = GeneratorCT2fromHfHub(