Skip to content

Commit

Permalink
try other way of getting torch
Browse files Browse the repository at this point in the history
  • Loading branch information
saraswatpuneet committed Apr 27, 2024
1 parent edf12bd commit d107c6b
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,7 @@
"tensorflow==2.14.0",
"transformers==4.36.0",
"unidecode==1.3.7",
"torch@https://download.pytorch.org/whl/cpu/torch-2.0.1%2Bcpu-cp38-cp38-linux_x86_64.whl"
]

# PyTorch index URL for dependency link
Expand All @@ -81,7 +82,7 @@

setup(
name="querent",
version="3.0.5",
version="3.0.6",
author="Querent AI",
description="The Asynchronous Data Dynamo and Graph Neural Network Catalyst",
long_description=long_description,
Expand Down Expand Up @@ -167,9 +168,6 @@
"Typing :: Typed",
],
python_requires=">=3.9, <=3.10",
dependency_links=[
f"{torch_index_url}/torch-2.0.1%2Bcpu-cp39-cp39-win_amd64.whl" # Adjust this accordingly
],
packages=find_packages(exclude=("tests", "tests.*")),
install_requires=requirements,
license="Business Source License 1.1",
Expand Down

0 comments on commit d107c6b

Please sign in to comment.