Skip to content

Commit

Permalink
Load models when model type changes
Browse files Browse the repository at this point in the history
  • Loading branch information
jhj0517 committed Nov 7, 2024
1 parent c0fafde commit 535b5e7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modules/live_portrait/live_portrait_inferencer.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def load_models(self,
)
self.stitching_retargeting_module = {"stitching": self.stitching_retargeting_module}

if self.pipeline is None:
if self.pipeline is None or model_type != self.model_type:
self.pipeline = LivePortraitWrapper(
InferenceConfig(),
self.appearance_feature_extractor,
Expand Down

0 comments on commit 535b5e7

Please sign in to comment.