Spaces:
Paused
Paused
Update pulid/pipeline.py
Browse files- pulid/pipeline.py +2 -0
pulid/pipeline.py
CHANGED
|
@@ -92,6 +92,8 @@ class PuLIDPipeline:
|
|
| 92 |
self.handler_ante = insightface.model_zoo.get_model('models/antelopev2/glintr100.onnx')
|
| 93 |
self.handler_ante.prepare(ctx_id=0)
|
| 94 |
|
|
|
|
|
|
|
| 95 |
gc.collect()
|
| 96 |
torch.cuda.empty_cache()
|
| 97 |
|
|
|
|
| 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')
|
| 96 |
+
|
| 97 |
gc.collect()
|
| 98 |
torch.cuda.empty_cache()
|
| 99 |
|