Update api/seedvr_server.py
Browse files- 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 |
-
"
|
| 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 = "
|
| 169 |
fps: Optional[float] = None,
|
| 170 |
-
debug: bool =
|
| 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:
|