Update aduc_framework/managers/seedvr_manager.py

aduc_framework/managers/seedvr_manager.py

@@ -109,16 +109,19 @@ class SeedVrManager:
         ckpt_dir.mkdir(exist_ok=True)
 
         pretrain_model_url = {
+            'configs_3b': 'https://huggingface.co/spaces/ByteDance-Seed/SeedVR2-3B/tree/main/configs_3b/main.yaml',
             'vae': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/ema_vae.pth',
             'dit': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/seedvr2_ema_3b.pth',
             'pos_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/pos_emb.pt',
             'neg_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt'
         }
-        
+
+        _load_file_from_url(url=pretrain_model_url['configs_3b'], model_dir='./ckpts/')
         _load_file_from_url(url=pretrain_model_url['dit'], model_dir='./ckpts/')
         _load_file_from_url(url=pretrain_model_url['vae'], model_dir='./ckpts/')
         _load_file_from_url(url=pretrain_model_url['pos_emb'])
         _load_file_from_url(url=pretrain_model_url['neg_emb'])
+        
         logger.info("Modelos do SeedVR2 baixados com sucesso.")
 
     def _initialize_runner(self):