Spaces:

yanze
/

PuLID

Running on Zero

yanze commited on May 1

Commit

e6c2ef9

•

1 Parent(s): 03f50c5

Update pulid/pipeline.py

Files changed (1) hide show

pulid/pipeline.py CHANGED Viewed

@@ -86,7 +86,7 @@ class PuLIDPipeline:
         # antelopev2
         snapshot_download('DIAMONIK7777/antelopev2', local_dir='models/antelopev2')
         self.app = FaceAnalysis(
-            name='antelopev2', root='.', providers=['CUDAExecutionProvider', 'CPUExecutionProvider']
         )
         self.app.prepare(ctx_id=0, det_size=(640, 640))
         self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx')

         # antelopev2
         snapshot_download('DIAMONIK7777/antelopev2', local_dir='models/antelopev2')
         self.app = FaceAnalysis(
+            name='antelopev2', root='.', providers=['CPUExecutionProvider']
         )
         self.app.prepare(ctx_id=0, det_size=(640, 640))
         self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx')