From 8f01f829d7b865fa1d4711e64f9f68cc877576a8 Mon Sep 17 00:00:00 2001
From: zzhhjjj <z785566960@gmail.com>
Date: Tue, 30 Apr 2024 13:41:18 +0000
Subject: [PATCH] add dependencies

---
 .github/workflows/llama_tests.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/llama_tests.yaml b/.github/workflows/llama_tests.yaml
index 56d55a8e..fe0e2557 100644
--- a/.github/workflows/llama_tests.yaml
+++ b/.github/workflows/llama_tests.yaml
@@ -1,4 +1,4 @@
-name: Run examples
+name: Run Llama example and loss test
 
 on:
   push:
@@ -43,9 +43,11 @@ jobs:
         python -m pip install --upgrade pip
         pip install packaging
         pip install wheel
+        pip install "flash-attn>=2.5.0"  --no-build-isolation
         pip install -e .
         pip install -e .[dev]
         pip install -e .[test]
+        pip install transformers datasets
 
     - name: Show installed libraries and their versions
       run: pip freeze | tee installed.txt