diff --git a/exo/download/hf/hf_helpers.py b/exo/download/hf/hf_helpers.py index a62dac0d..1ac34f31 100644 --- a/exo/download/hf/hf_helpers.py +++ b/exo/download/hf/hf_helpers.py @@ -118,7 +118,7 @@ async def move_models_to_hf(seed_dir: Union[str, Path]): print(f'Error moving model to .cache: {e}') - + async def fetch_file_list(session, repo_id, revision, path=""): api_url = f"{get_hf_endpoint()}/api/models/{repo_id}/tree/{revision}" url = f"{api_url}/{path}" if path else api_url diff --git a/scripts/build_exo.py b/scripts/build_exo.py index 67829b44..78a61249 100644 --- a/scripts/build_exo.py +++ b/scripts/build_exo.py @@ -55,4 +55,4 @@ def run(): print(f"An error occurred: {e}") if __name__ == "__main__": - run() \ No newline at end of file + run() diff --git a/setup.py b/setup.py index eb0053f5..2b504f25 100644 --- a/setup.py +++ b/setup.py @@ -13,7 +13,7 @@ install_requires = [ "Jinja2==3.1.4", "netifaces==0.11.0", "numpy==2.0.0", - "nuitka==2.4.11", + "nuitka==2.5.1", "nvidia-ml-py==12.560.30", "pillow==10.4.0", "prometheus-client==0.20.0", @@ -24,7 +24,7 @@ install_requires = [ "rich==13.7.1", "tenacity==9.0.0", "tqdm==4.66.4", - "transformers==4.46.3" if (sys.version_info.major==3 and sys.version_info.minor>12) else "transformers==4.43.3" , + "transformers==4.46.3", "uuid==1.30", "tinygrad @ git+https://github.com/tinygrad/tinygrad.git@232edcfd4f8b388807c64fb1817a7668ce27cbad", ]