aducsdr commited on
Commit
05f2657
·
verified ·
1 Parent(s): a612208

Update aduc_framework/managers/seedvr_manager.py

Browse files
aduc_framework/managers/seedvr_manager.py CHANGED
@@ -92,7 +92,7 @@ class SeedVrManager:
92
  model_urls = {
93
  'vae': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/ema_vae.pth',
94
  'dit_3b': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/seedvr2_ema_3b.pth',
95
- 'dit_7b': 'https://huggingface.co/ByteDance-Seed/SeedVR2-7B/resolve/main/seedvr2_ema_7b.pth',
96
  'pos_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/pos_emb.pt',
97
  'neg_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt'
98
  }
@@ -210,7 +210,7 @@ class SeedVrPlaceholder:
210
 
211
  try:
212
  with open("config.yaml", 'r') as f: config = yaml.safe_load(f)
213
- seedvr_gpus_required = config['specialists'].get('seedvr', {}).get('gpus_required', 0)
214
 
215
  if seedvr_gpus_required > 0:
216
  seedvr_device_ids = hardware_manager.allocate_gpus('SeedVR', seedvr_gpus_required)
 
92
  model_urls = {
93
  'vae': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/ema_vae.pth',
94
  'dit_3b': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/seedvr2_ema_3b.pth',
95
+ #'dit_7b': 'https://huggingface.co/ByteDance-Seed/SeedVR2-7B/resolve/main/seedvr2_ema_7b.pth',
96
  'pos_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/pos_emb.pt',
97
  'neg_emb': 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt'
98
  }
 
210
 
211
  try:
212
  with open("config.yaml", 'r') as f: config = yaml.safe_load(f)
213
+ seedvr_gpus_required = config['specialists'].get('seedvr', {}).get('gpus_required', 2)
214
 
215
  if seedvr_gpus_required > 0:
216
  seedvr_device_ids = hardware_manager.allocate_gpus('SeedVR', seedvr_gpus_required)