adding torch source

Author: thunderock

pyproject.toml

@@ -36,6 +36,12 @@ traittypes = "^0.2.1"
 traitlets = "^5.9.0"
 datasets = "^2.17.1"
 
+[[tool.poetry.source]]
+name = "torch-wheels"
+url = "https://data.pyg.org/whl/torch-2.1.2+cu121.html"
+default = false
+secondary = false
+
 [build-system]
 requires = ["poetry_core>=1.9.0"]
 build-backend = "poetry.core.masonry.api"