Update aduc_framework/managers/seedvr_manager.py
Browse files
aduc_framework/managers/seedvr_manager.py
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import torch
|
| 2 |
import imageio
|
| 3 |
import os
|
|
@@ -49,6 +52,8 @@ def setup_dependencies():
|
|
| 49 |
|
| 50 |
setup_dependencies()
|
| 51 |
|
|
|
|
|
|
|
| 52 |
# Função auxiliar para download
|
| 53 |
def _load_file_from_url(url, model_dir='./', file_name=None):
|
| 54 |
os.makedirs(model_dir, exist_ok=True)
|
|
@@ -59,6 +64,8 @@ def _load_file_from_url(url, model_dir='./', file_name=None):
|
|
| 59 |
download_url_to_file(url, cached_file, hash_prefix=None, progress=True)
|
| 60 |
return cached_file
|
| 61 |
|
|
|
|
|
|
|
| 62 |
# --- Importações diretas, assumindo que as pastas estão na raiz ---
|
| 63 |
from projects.video_diffusion_sr.infer import VideoDiffusionInfer
|
| 64 |
from common.config import load_config
|
|
@@ -71,6 +78,7 @@ from torchvision.transforms import Compose, Lambda, Normalize
|
|
| 71 |
from torchvision.io.video import read_video
|
| 72 |
from einops import rearrange
|
| 73 |
|
|
|
|
| 74 |
|
| 75 |
class SeedVrManager:
|
| 76 |
"""
|
|
@@ -220,14 +228,5 @@ class SeedVrManager:
|
|
| 220 |
finally:
|
| 221 |
self._unload_runner()
|
| 222 |
|
| 223 |
-
# Instância
|
| 224 |
-
def __init__(self, workspace_dir="deformes_workspace"):
|
| 225 |
-
self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 226 |
-
self.runner = None
|
| 227 |
-
self.workspace_dir = workspace_dir
|
| 228 |
-
self.is_initialized = False
|
| 229 |
-
self._original_barrier = None
|
| 230 |
-
self.setup_complete = False # Flag para rodar o setup apenas uma vez
|
| 231 |
-
logger.info("SeedVrManager initialized. Setup will run on first use.")
|
| 232 |
-
|
| 233 |
seedvr_manager_singleton = SeedVrManager()
|
|
|
|
| 1 |
+
# hd_specialist.py (Versão Final - Estrutura de Arquivos Corrigida)
|
| 2 |
+
#https://huggingface.co/spaces/ByteDance-Seed/SeedVR2-3B
|
| 3 |
+
|
| 4 |
import torch
|
| 5 |
import imageio
|
| 6 |
import os
|
|
|
|
| 52 |
|
| 53 |
setup_dependencies()
|
| 54 |
|
| 55 |
+
|
| 56 |
+
|
| 57 |
# Função auxiliar para download
|
| 58 |
def _load_file_from_url(url, model_dir='./', file_name=None):
|
| 59 |
os.makedirs(model_dir, exist_ok=True)
|
|
|
|
| 64 |
download_url_to_file(url, cached_file, hash_prefix=None, progress=True)
|
| 65 |
return cached_file
|
| 66 |
|
| 67 |
+
|
| 68 |
+
|
| 69 |
# --- Importações diretas, assumindo que as pastas estão na raiz ---
|
| 70 |
from projects.video_diffusion_sr.infer import VideoDiffusionInfer
|
| 71 |
from common.config import load_config
|
|
|
|
| 78 |
from torchvision.io.video import read_video
|
| 79 |
from einops import rearrange
|
| 80 |
|
| 81 |
+
|
| 82 |
|
| 83 |
class SeedVrManager:
|
| 84 |
"""
|
|
|
|
| 228 |
finally:
|
| 229 |
self._unload_runner()
|
| 230 |
|
| 231 |
+
# Instância Singleton
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 232 |
seedvr_manager_singleton = SeedVrManager()
|