Spaces:
Running
Running
Commit
·
e70447d
1
Parent(s):
1797915
fix gpu dep conflicts
Browse files
app.py
CHANGED
@@ -5,6 +5,8 @@ if nvidia_available:
|
|
5 |
os.system(
|
6 |
"pip install optimum-benchmark[onnxruntime-gpu,openvino,neural-compressor,diffusers,peft]@git+https://github.com/huggingface/optimum-benchmark.git"
|
7 |
)
|
|
|
|
|
8 |
DEVICES = ["cpu", "cuda"]
|
9 |
else:
|
10 |
os.system(
|
|
|
5 |
os.system(
|
6 |
"pip install optimum-benchmark[onnxruntime-gpu,openvino,neural-compressor,diffusers,peft]@git+https://github.com/huggingface/optimum-benchmark.git"
|
7 |
)
|
8 |
+
os.system("pip uninstall onnxruntime onnxruntime-gpu -y")
|
9 |
+
os.system("pip install onnxruntime-gpu")
|
10 |
DEVICES = ["cpu", "cuda"]
|
11 |
else:
|
12 |
os.system(
|