iofu728 commited on
Commit
9065e38
1 Parent(s): 9ecbf84

Fix(MInference): fix requirments

Browse files
Files changed (2) hide show
  1. app.py +4 -0
  2. requirements.txt +1 -2
app.py CHANGED
@@ -5,6 +5,10 @@ subprocess.run(
5
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
  shell=True,
7
  )
 
 
 
 
8
 
9
  import gradio as gr
10
  import os
 
5
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
  shell=True,
7
  )
8
+ subprocess.run(
9
+ "pip install pycuda==2023.1",
10
+ shell=True,
11
+ )
12
 
13
  import gradio as gr
14
  import os
requirements.txt CHANGED
@@ -2,5 +2,4 @@ triton==2.1.0
2
  accelerate
3
  transformers
4
  wheel
5
- setuptools
6
- pycuda==2023.1
 
2
  accelerate
3
  transformers
4
  wheel
5
+ setuptools