YinuoGuo27 commited on
Commit
ea49e7a
·
verified ·
1 Parent(s): eabcbc1

Update difpoint/inference.py

Browse files
Files changed (1) hide show
  1. difpoint/inference.py +1 -2
difpoint/inference.py CHANGED
@@ -273,7 +273,7 @@ class Inferencer(object):
273
  # 2024.06.26
274
  @torch.no_grad()
275
  def generate_with_audio_img(self, upload_audio_path, tts_audio_path, audio_type, image_path, smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t, save_path='results'):
276
-
277
  if audio_type == 'upload':
278
  audio_path = upload_audio_path
279
  elif audio_type == 'tts':
@@ -485,5 +485,4 @@ class Inferencer(object):
485
  print('#' * 25 + f'End Inference, cost time {time.time() - st}' + '#' * 25)
486
  return save_path
487
 
488
-
489
 
 
273
  # 2024.06.26
274
  @torch.no_grad()
275
  def generate_with_audio_img(self, upload_audio_path, tts_audio_path, audio_type, image_path, smoothed_pitch, smoothed_yaw, smoothed_roll, smoothed_t, save_path='results'):
276
+ print(audio_type)
277
  if audio_type == 'upload':
278
  audio_path = upload_audio_path
279
  elif audio_type == 'tts':
 
485
  print('#' * 25 + f'End Inference, cost time {time.time() - st}' + '#' * 25)
486
  return save_path
487
 
 
488