aducsdr commited on
Commit
617831b
·
verified ·
1 Parent(s): 9f7d882

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 Singletonclass SeedVrManager:
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()