Hugging Face's logo

Spaces:
Alpha-VLLM
/
Lumina-Image-2.0
Running on Zero

App Files Community
6
Dakerqi commited on
Commit
7446374
·
verified ·
1 Parent(s): 7af10a5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -10,5 +10,5 @@ gradio
10
  torchdiffeq
11
  click
12
  torchvision==0.16.0
13
- pytorch-cuda==12.1
14
  flash_attn==2.5.9.post1
 
10
  torchdiffeq
11
  click
12
  torchvision==0.16.0
13
+ cudatoolkit==12.1
14
  flash_attn==2.5.9.post1