jhj0517 commited on
Commit
329d0a4
1 Parent(s): bb6e7ec

keep model after inference

Browse files
Files changed (1) hide show
  1. pose_align.py +0 -1
pose_align.py CHANGED
@@ -310,7 +310,6 @@ class PoseAlignmentInference:
310
  clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(result_pose_only, fps=fps)
311
  clip.write_videofile(outfn_align_pose_video, fps=fps)
312
  print('pose align done')
313
- self.release_vram()
314
  return outfn_align_pose_video, outfn
315
 
316
  def init_model(self):
 
310
  clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(result_pose_only, fps=fps)
311
  clip.write_videofile(outfn_align_pose_video, fps=fps)
312
  print('pose align done')
 
313
  return outfn_align_pose_video, outfn
314
 
315
  def init_model(self):