Spaces:

yanze
/

PuLID

Running on Zero

yanze commited on May 1

Commit

01bfd61

•

1 Parent(s): e6c2ef9

Update pulid/pipeline.py

Files changed (1) hide show

pulid/pipeline.py CHANGED Viewed

@@ -92,6 +92,8 @@ class PuLIDPipeline:
         self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx')
         self.handler_ante.prepare(ctx_id=0)
         gc.collect()
         torch.cuda.empty_cache()

         self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx')
         self.handler_ante.prepare(ctx_id=0)
+        print('load done')
         gc.collect()
         torch.cuda.empty_cache()