jhj0517
commited on
Commit
·
535b5e7
1
Parent(s):
c0fafde
Load models when model type changes
Browse files
modules/live_portrait/live_portrait_inferencer.py
CHANGED
|
@@ -119,7 +119,7 @@ class LivePortraitInferencer:
|
|
| 119 |
)
|
| 120 |
self.stitching_retargeting_module = {"stitching": self.stitching_retargeting_module}
|
| 121 |
|
| 122 |
-
if self.pipeline is None:
|
| 123 |
self.pipeline = LivePortraitWrapper(
|
| 124 |
InferenceConfig(),
|
| 125 |
self.appearance_feature_extractor,
|
|
|
|
| 119 |
)
|
| 120 |
self.stitching_retargeting_module = {"stitching": self.stitching_retargeting_module}
|
| 121 |
|
| 122 |
+
if self.pipeline is None or model_type != self.model_type:
|
| 123 |
self.pipeline = LivePortraitWrapper(
|
| 124 |
InferenceConfig(),
|
| 125 |
self.appearance_feature_extractor,
|