yanze commited on
Commit
dd561ba
1 Parent(s): 01bfd61

Update pulid/pipeline.py

Browse files
Files changed (1) hide show
  1. pulid/pipeline.py +1 -1
pulid/pipeline.py CHANGED
@@ -89,7 +89,7 @@ class PuLIDPipeline:
89
  name='antelopev2', root='.', providers=['CPUExecutionProvider']
90
  )
91
  self.app.prepare(ctx_id=0, det_size=(640, 640))
92
- self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx')
93
  self.handler_ante.prepare(ctx_id=0)
94
 
95
  print('load done')
 
89
  name='antelopev2', root='.', providers=['CPUExecutionProvider']
90
  )
91
  self.app.prepare(ctx_id=0, det_size=(640, 640))
92
+ self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx', providers=['CPUExecutionProvider'])
93
  self.handler_ante.prepare(ctx_id=0)
94
 
95
  print('load done')