jhj0517 commited on
Commit
dc1090e
·
unverified ·
2 Parent(s): c0fafde 535b5e7

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,