- handler.py +3 -0
handler.py
CHANGED
|
@@ -3,6 +3,9 @@ warnings.filterwarnings('ignore')
|
|
| 3 |
|
| 4 |
import subprocess, io, os, sys, time
|
| 5 |
|
|
|
|
|
|
|
|
|
|
| 6 |
run_gradio = False
|
| 7 |
if os.environ.get('IS_MY_DEBUG') is None:
|
| 8 |
run_gradio = True
|
|
|
|
| 3 |
|
| 4 |
import subprocess, io, os, sys, time
|
| 5 |
|
| 6 |
+
is_production = True
|
| 7 |
+
os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
|
| 8 |
+
|
| 9 |
run_gradio = False
|
| 10 |
if os.environ.get('IS_MY_DEBUG') is None:
|
| 11 |
run_gradio = True
|