fix error2

Author: daikang6

.github/workflows/e2e_ppo_trainer_megatron_vllm_2_ascend.yml

@@ -213,6 +213,7 @@ jobs:
         run: |
           pip install -r requirements-npu.txt
           pip install --no-deps -e .
+          pip install trl
       - name: Check final pip list
         run: |
           pip list

tests/special_e2e/run_transferqueue.sh

@@ -167,6 +167,15 @@ print(get_device_name())
 EOF
 )
 
+    extra_flash_args=()
+
+    if [ "$device_name" == "npu" ]; then
+        echo "Detect NPU device, enabling FlashAttention..."
+        extra_flash_args+=(
+            ++actor_rollout_ref.actor.megatron.override_transformer_config.use_flash_attn=True
+        )
+    fi
+
     # For Ascend NPU, please add:
     #++actor_rollout_ref.actor.megatron.override_transformer_config.use_flash_attn=True \
     #++actor_rollout_ref.ref.megatron.override_transformer_config.use_flash_attn=True \