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',
|
| 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)
|