Eueuiaa commited on
Commit
94cdf7d
·
verified ·
1 Parent(s): 321fc5a

Update api/seedvr_server.py

Browse files
Files changed (1) hide show
  1. api/seedvr_server.py +3 -3
api/seedvr_server.py CHANGED
@@ -146,7 +146,7 @@ class SeedVRServer:
146
  # Baixa os checkpoints do Hugging Face se não existirem
147
  print(f"[SeedVRServer] Verificando checkpoints em {self.CKPTS_ROOT}...")
148
  model_files = {
149
- "seedvr2_ema_3b_fp16.safetensors": "MonsterMMORPG/SeedVR2_SECourses",
150
  "ema_vae_fp16.safetensors": "MonsterMMORPG/SeedVR2_SECourses"
151
  }
152
  for filename, repo_id in model_files.items():
@@ -165,9 +165,9 @@ class SeedVRServer:
165
  seed: int,
166
  resolution: int,
167
  batch_size: int,
168
- model: str = "seedvr2_ema_3b_fp16.safetensors",
169
  fps: Optional[float] = None,
170
- debug: bool = False,
171
  preserve_vram: bool = True,
172
  progress: Optional[Callable] = None
173
  ) -> str:
 
146
  # Baixa os checkpoints do Hugging Face se não existirem
147
  print(f"[SeedVRServer] Verificando checkpoints em {self.CKPTS_ROOT}...")
148
  model_files = {
149
+ "seedvr2_ema_7b_sharp_fp16.safetensors": "MonsterMMORPG/SeedVR2_SECourses",
150
  "ema_vae_fp16.safetensors": "MonsterMMORPG/SeedVR2_SECourses"
151
  }
152
  for filename, repo_id in model_files.items():
 
165
  seed: int,
166
  resolution: int,
167
  batch_size: int,
168
+ model: str = "seedvr2_ema_7b_sharp_fp16.safetensors",
169
  fps: Optional[float] = None,
170
+ debug: bool = True,
171
  preserve_vram: bool = True,
172
  progress: Optional[Callable] = None
173
  ) -> str: