Spaces:

openfree
/

ginigen-sora

Running

benibraz commited on Nov 21, 2024

Commit

714b7fb

1 Parent(s): 7bb8767

Remove unnecessary pipeline.to("cpu") calls in video generation functions

Files changed (1) hide show

app.py CHANGED Viewed

@@ -224,7 +224,6 @@ def generate_video_from_text(
             duration=5,
         )
     finally:
-        pipeline.to("cpu")
         torch.cuda.empty_cache()
         gc.collect()
@@ -322,7 +321,6 @@ def generate_video_from_image(
         )
     finally:
-        pipeline.to("cpu")
         torch.cuda.empty_cache()
         gc.collect()

             duration=5,
         )
     finally:
         torch.cuda.empty_cache()
         gc.collect()
         )
     finally:
         torch.cuda.empty_cache()
         gc.collect()