From 75adf4035b0325064bb94e86ec4760ab3639c4d6 Mon Sep 17 00:00:00 2001 From: Lu Zhang Date: Wed, 28 Jun 2023 15:28:05 +0800 Subject: [PATCH] add requirements.txt for Dockerfile --- requirements.txt | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..f9250dbc --- /dev/null +++ b/requirements.txt @@ -0,0 +1,5 @@ +packaging +PyYAML==5.4.1 +deepspeed==0.6.5 +flash-attn==1.0.2 +bminf