iofu728 commited on
Commit
1c322b1
1 Parent(s): 4a12906

Feature(MInference): add pycuda==2023.1

Browse files
Files changed (2) hide show
  1. app.py +4 -0
  2. requirements.txt +0 -1
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
@@ -1,4 +1,3 @@
1
  triton==2.1.0
2
- pycuda==2023.1
3
  accelerate
4
  transformers
 
1
  triton==2.1.0
 
2
  accelerate
3
  transformers