jhj0517 commited on
Commit
284f5ff
·
1 Parent(s): 9fa70f5

fix frame bug

Browse files
modules/live_portrait/live_portrait_inferencer.py CHANGED
@@ -277,10 +277,10 @@ class LivePortraitInferencer:
277
  self.driving_values = self.prepare_driving_video(driving_images)
278
  driving_length = len(self.driving_values)
279
 
280
- total_length = max(driving_length, src_length)
281
-
282
  if animate_without_vid:
283
  total_length = total_length
 
284
  self.psi_list = [self.psi_list[0] for _ in range(total_length)]
285
 
286
  c_i_es = ExpressionSet()
 
277
  self.driving_values = self.prepare_driving_video(driving_images)
278
  driving_length = len(self.driving_values)
279
 
280
+ total_length = len(driving_images)
 
281
  if animate_without_vid:
282
  total_length = total_length
283
+
284
  self.psi_list = [self.psi_list[0] for _ in range(total_length)]
285
 
286
  c_i_es = ExpressionSet()