update setup.py

michaelfeil · May 19, 2023 · 23e683e · 23e683e
1 parent 66b1ead
commit 23e683e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/hf_hub_ctranslate2/__init__.py b/hf_hub_ctranslate2/__init__.py
@@ -2,4 +2,5 @@
 """Compatability between Huggingface and Ctranslate2."""
 # __all__ = ["__version__", "TranslatorCT2fromHfHub", "GeneratorCT2fromHfHub", "MultiLingualTranslatorCT2fromHfHub", "_private"]
 from hf_hub_ctranslate2.translate import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub, MultiLingualTranslatorCT2fromHfHub
+from hf_hub_ctranslate2.util import _download_model
 __version__ = "2.0.6"
diff --git a/hf_hub_ctranslate2/translate.py b/hf_hub_ctranslate2/translate.py
@@ -14,7 +14,7 @@
 from typing import Any, Union, List
 import os
 
-from hf_hub_ctranslate2.util import _download_model
+from hf_hub_ctranslate2.util import utils as _utils
 
 
 class CTranslate2ModelfromHuggingfaceHub:
@@ -34,10 +34,10 @@ def __init__(
             model_path = model_name_or_path
         else:
             try:
-                model_path = _download_model(model_name_or_path, hub_kwargs=hub_kwargs)
+                model_path = _utils._download_model(model_name_or_path, hub_kwargs=hub_kwargs)
             except:
                 hub_kwargs["local_files_only"] = True
-                model_path = _download_model(model_name_or_path, hub_kwargs=hub_kwargs)
+                model_path = _utils._download_model(model_name_or_path, hub_kwargs=hub_kwargs)
         self.model = self.ctranslate_class(
             model_path,
             device=device,

diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@
 
 setup(
     name="hf_hub_ctranslate2",
-    packages=find_packages(where="hf_hub_ctranslate2"),
+    packages=find_packages(),
     version=version,
     description=("Connecting Transfromers on HuggingfaceHub with Ctranslate2 "),
     long_description=long_description,