📌 Use cpu version of torch

Author: pajowu

pyproject.toml

@@ -140,6 +140,15 @@ production = [
   "pillow-avif-plugin",
 ]
 
+[tool.uv.sources]
+torch = { index = "pytorch-cpu" }
+torchvision = { index = "pytorch-cpu" }
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
 [build-system]
 requires = ["setuptools"]
 build-backend = "setuptools.build_meta"

requirements-dev.txt

@@ -18,9 +18,7 @@ asgiref==3.8.1
     #   django
     #   django-countries
 async-timeout==4.0.3
-    # via
-    #   fragdenstaat-de (pyproject.toml)
-    #   aiohttp
+    # via fragdenstaat-de (pyproject.toml)
 attrs==24.2.0
     # via
     #   aiohttp
@@ -422,8 +420,6 @@ enmerkar==0.7.1
     # via django-prices
 et-xmlfile==1.1.0
     # via openpyxl
-exceptiongroup==1.2.2
-    # via pytest
 factory-boy==3.3.1
     # via
     #   fragdenstaat-de (pyproject.toml)
@@ -860,6 +856,7 @@ setuptools==74.1.2
     #   django-cms
     #   geoip2
     #   ipython
+    #   torch
 six==1.16.0
     # via
     #   bleach
@@ -912,12 +909,6 @@ tinycss2==1.3.0
     #   cssselect2
     #   svglib
     #   weasyprint
-tomli==2.1.0
-    # via
-    #   djlint
-    #   ipdb
-    #   pytest
-    #   sphinx
 torch==2.5.1
     # via
     #   fragdenstaat-de (pyproject.toml)
@@ -932,15 +923,12 @@ traitlets==5.14.3
     #   matplotlib-inline
 typing-extensions==4.12.2
     # via
-    #   asgiref
     #   dj-database-url
     #   django-countries
     #   elasticsearch-dsl
     #   jwcrypto
-    #   multidict
     #   psycopg
     #   pyee
-    #   pypdf
     #   pytest-factoryboy
     #   python-docx
     #   torch

requirements-production.txt

@@ -15,10 +15,7 @@ asgiref==3.8.1
     #   django
     #   django-countries
 async-timeout==4.0.3
-    # via
-    #   fragdenstaat-de (pyproject.toml)
-    #   aiohttp
-    #   redis
+    # via fragdenstaat-de (pyproject.toml)
 attrs==24.2.0
     # via
     #   aiohttp
@@ -732,6 +729,7 @@ setuptools==74.1.2
     # via
     #   django-cms
     #   geoip2
+    #   torch
 six==1.16.0
     # via
     #   bleach
@@ -768,17 +766,13 @@ torchvision==0.20.1
     # via fragdenstaat-de (pyproject.toml)
 typing-extensions==4.12.2
     # via
-    #   asgiref
     #   dj-database-url
     #   django-countries
     #   elasticsearch-dsl
     #   jwcrypto
-    #   multidict
     #   psycopg
-    #   pypdf
     #   python-docx
     #   torch
-    #   uvicorn
 tzdata==2024.1
     # via
     #   celery

requirements.txt

@@ -14,9 +14,7 @@ asgiref==3.8.1
     #   django
     #   django-countries
 async-timeout==4.0.3
-    # via
-    #   fragdenstaat-de (pyproject.toml)
-    #   aiohttp
+    # via fragdenstaat-de (pyproject.toml)
 attrs==23.2.0
     # via
     #   aiohttp
@@ -713,6 +711,7 @@ setuptools==74.1.2
     # via
     #   django-cms
     #   geoip2
+    #   torch
 six==1.16.0
     # via
     #   bleach
@@ -749,14 +748,11 @@ torchvision==0.20.1
     # via fragdenstaat-de (pyproject.toml)
 typing-extensions==4.12.2
     # via
-    #   asgiref
     #   dj-database-url
     #   django-countries
     #   elasticsearch-dsl
     #   jwcrypto
-    #   multidict
     #   psycopg
-    #   pypdf
     #   python-docx
     #   torch
 tzdata==2024.1