Update

[ghstack-poisoned]

Author: vmoens

.github/unittest/linux_libs/scripts_llm/environment.yml renamed to .github/unittest/llm/scripts_llm/environment.yml

@@ -23,14 +23,4 @@ lib_dir="${env_dir}/lib"
 
 conda deactivate && conda activate ./env
 
-python -c "import transformers, datasets"
-
-pytest test/test_rlhf.py --instafail -v --durations 200 --capture no --error-for-skips
-
-python examples/rlhf/train_rlhf.py \
-  sys.device=cuda:0 sys.ref_device=cuda:0 \
-  model.name_or_path=gpt2 train.max_epochs=2 \
-  data.batch_size=2 train.ppo.ppo_batch_size=2 \
-  train.ppo.ppo_num_epochs=1 reward_model.name_or_path= \
-  train.ppo.episode_length=8 train.ppo.num_rollouts_per_epoch=4 \
-  data.block_size=110 io.logger=csv
+pytest test/llm -vvv --instafail --durations 600 --capture no --error-for-skips

.github/unittest/linux_libs/scripts_llm/install.sh renamed to .github/unittest/llm/scripts_llm/install.sh

@@ -21,6 +21,7 @@ permissions:
 
 jobs:
   unittests:
+    if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'llm/') }}
     strategy:
       matrix:
         python_version: ["3.9"]