YinuoGuo27 commited on
Commit
d49e46e
·
verified ·
1 Parent(s): 0ebdef8

Update difpoint/src/models/predictor.py

Browse files
Files changed (1) hide show
  1. difpoint/src/models/predictor.py +1 -1
difpoint/src/models/predictor.py CHANGED
@@ -173,7 +173,7 @@ class OnnxRuntimePredictor:
173
  """
174
  OnnxRuntime Prediction
175
  """
176
- @spaces.GPU
177
  def __init__(self, **kwargs):
178
  model_path = kwargs.get("model_path", "") # 用模型路径区分是否是一样的实例
179
  assert os.path.exists(model_path), "model path must exist!"
 
173
  """
174
  OnnxRuntime Prediction
175
  """
176
+
177
  def __init__(self, **kwargs):
178
  model_path = kwargs.get("model_path", "") # 用模型路径区分是否是一样的实例
179
  assert os.path.exists(model_path), "model path must exist!"