Keep torch at 2.2.2 for macos x86 versions

2 years ago · 1b66ad4603
parent ac5e687c8f
commit 1b66ad4603
1 changed files with 2 additions and 1 deletions
--- a/requirements-freqai-rl.txt
+++ b/requirements-freqai-rl.txt
@ -2,7 +2,8 @@
 -r requirements-freqai.txt

 # Required for freqai-rl
-torch==2.3.1
+torch==2.3.1; sys_platform != 'darwin' or platform_machine != 'x86_64'
+torch==2.2.2; sys_platform == 'darwin' and platform_machine == 'x86_64'
 gymnasium==0.29.1
 stable_baselines3==2.3.2
 sb3_contrib>=2.2.1