more...

Borda · Borda · commit 0c1e08700580 · 2025-01-09T15:23:02.000+09:00
diff --git a/lightning_examples/reinforce-learning-DQN/requirements.txt b/lightning_examples/reinforce-learning-DQN/requirements.txt
@@ -1,7 +1,10 @@
 gym <0.24
-pygame <=2.7.0
+pygame
 pandas
 seaborn
 # todo: pin version intill reinstall with PT-eco alignment
 torch ==2.1.*
 torchvision ==0.16.*
+
+# todo: some compatibility issues
+numpy <2.0
diff --git a/lightning_examples/text-transformers/requirements.txt b/lightning_examples/text-transformers/requirements.txt
@@ -2,8 +2,8 @@ scipy
 scikit-learn
 torchmetrics >=1.0, <1.2
 torchtext
-transformers <=4.44.0
-datasets
+transformers
+datasets <=2.21
 # todo: pin version intill reinstall with PT-eco alignment
 torch ==2.1.*
 torchvision ==0.16.*
