thunderock
diff --git a/‎poetry.lock
+1-1 b/‎poetry.lock
+1-1
diff --git a/‎pyproject.toml
+5 b/‎pyproject.toml
+5
@@ -33,6 +33,11 @@ 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"
+priority = "primary"
+
 [build-system]
 requires = ["poetry_core>=1.9.0"]
 build-backend = "poetry.core.masonry.api"