Daniellesry commited on
Commit
1dbdebb
·
1 Parent(s): 9d09b4a
Files changed (1) hide show
  1. dkt/pipelines/pipeline.py +1 -1
dkt/pipelines/pipeline.py CHANGED
@@ -961,7 +961,7 @@ class DKTPipeline:
961
 
962
  del vae_outs
963
  # video = video.cpu()
964
- torch.cuda.empty_cache()
965
 
966
 
967
  processed_video = video[:frame_length]
 
961
 
962
  del vae_outs
963
  # video = video.cpu()
964
+ # torch.cuda.empty_cache()
965
 
966
 
967
  processed_video = video[:frame_length]