jhj0517 commited on
Commit
2ce5fe4
·
1 Parent(s): 8db15a4

Fix None error

Browse files
modules/live_portrait/live_portrait_inferencer.py CHANGED
@@ -185,7 +185,7 @@ class LivePortraitInferencer:
185
  es.r = self.calc_fe(es.e, blink, eyebrow, wink, pupil_x, pupil_y, aaa, eee, woo, smile,
186
  rotate_pitch, rotate_yaw, rotate_roll)
187
 
188
- if add_exp is not None:
189
  es.add(add_exp)
190
 
191
  new_rotate = get_rotation_matrix(s_info['pitch'] + es.r[0], s_info['yaw'] + es.r[1],
 
185
  es.r = self.calc_fe(es.e, blink, eyebrow, wink, pupil_x, pupil_y, aaa, eee, woo, smile,
186
  rotate_pitch, rotate_yaw, rotate_roll)
187
 
188
+ if isinstance(add_exp, ExpressionSet):
189
  es.add(add_exp)
190
 
191
  new_rotate = get_rotation_matrix(s_info['pitch'] + es.r[0], s_info['yaw'] + es.r[1],