From 23e683e48e287f86dd06c1ec1eb49bcb7a6d192b Mon Sep 17 00:00:00 2001 From: Michael Feil Date: Fri, 19 May 2023 02:01:33 +0200 Subject: [PATCH] update setup.py --- hf_hub_ctranslate2/__init__.py | 1 + hf_hub_ctranslate2/translate.py | 6 +++--- setup.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/hf_hub_ctranslate2/__init__.py b/hf_hub_ctranslate2/__init__.py index 09c557f..cc8dfef 100644 --- 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" \ No newline at end of file diff --git a/hf_hub_ctranslate2/translate.py b/hf_hub_ctranslate2/translate.py index 825f686..26e31ce 100644 --- 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 index 4c66070..b97c5f7 100644 --- 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,