Skip to content

Commit 1b253ca

Browse files
author
chibu
committed
pin torch to 2.7.0
1 parent 6e482f9 commit 1b253ca

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/automation/tasks/arenahard_generate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
class ArenaHardGenerateTask(BaseTask):
1111

1212
arenahard_packages = [
13+
"torch==2.7.0",
1314
"transformers==4.48.1",
1415
#"transformers==4.53.0",
1516
"huggingface-hub==0.34.3",

0 commit comments

Comments
 (0)