vinthony commited on
Commit
fa7bc69
1 Parent(s): 2ad9d5d

Update modules/sadtalker_test.py

Browse files
Files changed (1) hide show
  1. modules/sadtalker_test.py +1 -1
modules/sadtalker_test.py CHANGED
@@ -86,7 +86,7 @@ class SadTalker():
86
  batch = get_data(first_coeff_path, audio_path, self.device)
87
  coeff_path = self.audio_to_coeff.generate(batch, save_dir, pose_style)
88
  #coeff2video
89
- batch_size = 4
90
  data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path, batch_size, still_mode=still_mode)
91
  self.animate_from_coeff.generate(data, save_dir, enhancer='gfpgan' if use_enhancer else None)
92
  video_name = data['video_name']
 
86
  batch = get_data(first_coeff_path, audio_path, self.device)
87
  coeff_path = self.audio_to_coeff.generate(batch, save_dir, pose_style)
88
  #coeff2video
89
+ batch_size = 8
90
  data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path, batch_size, still_mode=still_mode)
91
  self.animate_from_coeff.generate(data, save_dir, enhancer='gfpgan' if use_enhancer else None)
92
  video_name = data['video_name']