Kunpeng Song commited on
Commit
0b29f0b
1 Parent(s): 942f8ec
Files changed (2) hide show
  1. .DS_Store +0 -0
  2. requirements.txt +2 -3
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
requirements.txt CHANGED
@@ -12,7 +12,7 @@ uvicorn
12
  wandb
13
  shortuuid
14
  httpx==0.24.0
15
- deepspeed==0.12.6
16
  peft==0.4.0
17
  transformers==4.36.2
18
  accelerate==0.21.0
@@ -29,5 +29,4 @@ torchaudio
29
  torchmetrics
30
  llava-torch
31
  rembg
32
- pytorch_lightning
33
- flash-attn
 
12
  wandb
13
  shortuuid
14
  httpx==0.24.0
15
+ deepspeed==0.13.1
16
  peft==0.4.0
17
  transformers==4.36.2
18
  accelerate==0.21.0
 
29
  torchmetrics
30
  llava-torch
31
  rembg
32
+ pytorch_lightning