Update app.py

app.py

@@ -11,14 +11,6 @@
 # // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # // See the License for the specific language governing permissions and
 # // limitations under the License.
-import spaces
-import subprocess
-
-# Clone the repository to ensure all files are available
-subprocess.run("git lfs install", shell=True, check=True)
-subprocess.run("git clone https://huggingface.co/spaces/ByteDance-Seed/SeedVR2-3B", shell=True, check=True)
-os.chdir('SeedVR2-3B')
-
 
 import os
 import torch
@@ -107,7 +99,7 @@ def configure_runner(sp_size):
         runner.vae.set_memory_limit(**runner.config.vae.memory_limit)
     return runner
 
-
+@torch.no_grad()
 def generation_step(runner, text_embeds_dict, cond_latents):
     def _move_to_cuda(x):
         return [i.to(torch.device("cuda")) for i in x]
@@ -164,7 +156,7 @@ def generation_step(runner, text_embeds_dict, cond_latents):
 
     return samples
 
-
+@torch.no_grad()
 def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size=1, cfg_scale=1.0, cfg_rescale=0.0, sample_steps=1, res_h=1280, res_w=720, sp_size=1):
     runner = configure_runner(1)