update

app.py

@@ -120,7 +120,7 @@ pretrain_model_url = {
     'neg_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt'
 }
 # download weights
-if not os.path.exists('./ckpts/seedvr_ema_3b.pth'):
+if not os.path.exists('./ckpts/seedvr2_ema_3b.pth'):
 	load_file_from_url(url=pretrain_model_url['dit'], model_dir='./ckpts/', progress=True, file_name=None)
 if not os.path.exists('./ckpts/ema_vae.pth'):
 	load_file_from_url(url=pretrain_model_url['vae'], model_dir='./ckpts/', progress=True, file_name=None)
@@ -153,7 +153,7 @@ def configure_runner(sp_size):
     
     # init_torch(cudnn_benchmark=False, timeout=datetime.timedelta(seconds=3600))
     # configure_sequence_parallel(sp_size)
-    runner.configure_dit_model(device="cuda", checkpoint='./ckpts/seedvr_ema_3b.pth')
+    runner.configure_dit_model(device="cuda", checkpoint='./ckpts/seedvr2_ema_3b.pth')
     runner.configure_vae_model()
     # Set memory limit.
     if hasattr(runner.vae, "set_memory_limit"):