TobDeBer commited on
Commit
8cb7a61
·
verified ·
1 Parent(s): 8b70209

Update app_onnx.py

Browse files
Files changed (1) hide show
  1. app_onnx.py +2 -2
app_onnx.py CHANGED
@@ -478,8 +478,8 @@ if __name__ == "__main__":
478
  ]
479
  }
480
  models = {}
481
- providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
482
- device = "cuda"
483
 
484
  for name, (repo_id, path, loras) in models_info.items():
485
  model_base_path = hf_hub_download_retry(repo_id=repo_id, filename=f"{path}onnx/model_base.onnx")
 
478
  ]
479
  }
480
  models = {}
481
+ providers = [CPUExecutionProvider']
482
+ device = "cpu"
483
 
484
  for name, (repo_id, path, loras) in models_info.items():
485
  model_base_path = hf_hub_download_retry(repo_id=repo_id, filename=f"{path}onnx/model_base.onnx")