Merge pull request #5 from jhj0517/feature/fix-model-loading
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,
|