Skip to content

Commit 44b9353

Browse files
committed
Use CUDA and update packages.
1 parent 3d65915 commit 44b9353

File tree

2 files changed

+420
-254
lines changed

2 files changed

+420
-254
lines changed

pyproject.toml

+15-14
Original file line numberDiff line numberDiff line change
@@ -1,35 +1,36 @@
11
[project]
2-
name = "transformers_for_sentiment_analysis"
2+
name = "transformers-for-sentiment-analysis"
33
version = "0.1.0"
44
readme = "README.md"
55
requires-python = ">=3.12"
66
dependencies = [
77
"datasets>=3.1.0",
88
"polars>=1.17.1",
9-
"torch>=2.5.1",
10-
"transformers>=4.47.0",
9+
"transformers[torch]>=4.47.0",
1110
]
1211

13-
[[tool.uv.index]]
14-
name = "pytorch-cpu"
15-
url = "https://download.pytorch.org/whl/cpu"
16-
explicit = true
17-
18-
[tool.uv.sources]
19-
torch = [{ index = "pytorch-cpu", marker = "platform_system != 'Darwin'" }]
12+
# [[tool.uv.index]]
13+
# name = "pytorch-cpu"
14+
# url = "https://download.pytorch.org/whl/cpu"
15+
# explicit = true
16+
#
17+
# [tool.uv.sources]
18+
# torch = [{ index = "pytorch-cpu", marker = "platform_system != 'Darwin'" }]
2019

2120
[tool.ruff.lint]
2221
select = ["ALL"]
2322
ignore = [
2423
# print is fine
2524
"T201",
26-
# conflicts with D211
27-
"D203",
28-
# conflicts with D213
29-
"D213",
3025
# documentation is hard
3126
"D100",
3227
"D103",
28+
# noise
29+
"TC002",
30+
# conflicts with D211
31+
"D203",
32+
# conflicts with D212
33+
"D213",
3334
# can't be used with formatter
3435
"COM812",
3536
"ISC001",

0 commit comments

Comments
 (0)