We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ada7dba commit 84c6f50Copy full SHA for 84c6f50
tests/test_translate.py
@@ -1,5 +1,5 @@
1
from hf_hub_ctranslate2 import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub, MultiLingualTranslatorCT2fromHfHub
2
-from hf_hub_ctranslate2.util import _download_model
+from hf_hub_ctranslate2.util import utils as _utils
3
from transformers import AutoTokenizer
4
5
@@ -51,7 +51,7 @@ def test_generator(model_name="michaelfeil/ct2fast-pythia-160m"):
51
52
53
def test_generator_single(model_name="michaelfeil/ct2fast-pythia-160m"):
54
- model_path = _download_model(model_name)
+ model_path = _utils._download_model(model_name)
55
tokenizer = AutoTokenizer.from_pretrained(model_path)
56
57
model = GeneratorCT2fromHfHub(
0 commit comments