ford442 commited on
Commit
3c07896
·
verified ·
1 Parent(s): d08a051

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ os.system(f'{conda_bin}/conda install nvidia/label/cuda-12.4.0::cuda-nvcc')
32
  #os.system(f'{conda_bin}/conda install pytorch::pytorch-cuda')
33
  #os.system(f'{conda_bin}/conda install rcdr_py37::tensorrt')
34
  #subprocess.run(['sh', './hidet.sh'])
 
35
  import hidet
36
  #print(dir(hidet))
37
  import torch_tensorrt
 
32
  #os.system(f'{conda_bin}/conda install pytorch::pytorch-cuda')
33
  #os.system(f'{conda_bin}/conda install rcdr_py37::tensorrt')
34
  #subprocess.run(['sh', './hidet.sh'])
35
+ subprocess.run(['sh', './modelopt.sh'])
36
  import hidet
37
  #print(dir(hidet))
38
  import torch_tensorrt