Spaces:
Running
on
Zero
Running
on
Zero
Update pulid/pipeline.py
Browse files- 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')
|