Skip to content

Commit 6f68da8

Browse files
committed
[Refactor] Remove LLM features for release
ghstack-source-id: f10f7a57e9140f5bf2544dbaeb5329891b151911 Pull Request resolved: #2912
1 parent 0475cbf commit 6f68da8

25 files changed

+19
-4389
lines changed

.github/unittest/linux_libs/scripts_llm/install.sh

100755100644
File mode changed.

.github/unittest/linux_libs/scripts_llm/post_process.sh

100755100644
File mode changed.

.github/unittest/linux_libs/scripts_llm/run-clang-format.py

100755100644
File mode changed.

.github/unittest/linux_libs/scripts_llm/run_test.sh

100755100644
-2
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,6 @@ python -c "import transformers, datasets"
2727

2828
pytest test/test_rlhf.py --instafail -v --durations 200 --capture no --error-for-skips
2929

30-
pytest test/test_actors.py test/test_collector.py -k llm --instafail -v --durations 200 --capture no --error-for-skips --runslow
31-
3230
pytest examples/rlhf/train_rlhf.py \
3331
sys.device=cuda:0 sys.ref_device=cuda:0 \
3432
model.name_or_path=gpt2 train.max_epochs=2 \

.github/unittest/linux_libs/scripts_llm/setup_env.sh

100755100644
File mode changed.

docs/source/reference/envs.rst

-1
Original file line numberDiff line numberDiff line change
@@ -521,7 +521,6 @@ TorchRL offers a series of custom built-in environments.
521521
ChessEnv
522522
PendulumEnv
523523
TicTacToeEnv
524-
LLMEnv
525524
LLMHashingEnv
526525

527526

0 commit comments

Comments
 (0)