Update api/seedvr_server.py

api/seedvr_server.py

@@ -63,38 +63,42 @@ class SeedVRServer:
         shutil.copy2(input_file, job_input_dir / Path(input_file).name)
         return job_input_dir, out_dir
         
-    def run_inference(self, filepath: str, *, seed: int, resh: int, resw: int, spsize: int, fps: Optional[float] = None):
+    def run_inference(self, file_path: str, *, seed: int, res_h: int, res_w: int, sp_size: int, fps: Optional[float] = None) -> Tuple[Optional[str], Optional[str], Path]:
         script = self.SEEDVR_ROOT / "inference_cli.py"
-        job_input_dir, outdir = self._prepare_job(filepath)
-        mediatype, _ = mimetypes.guess_type(filepath)
-        is_image = mediatype and mediatype.startswith("image")
-    
+        job_input_dir, out_dir = self._prepare_job(file_path)
+
+        media_type, _ = mimetypes.guess_type(file_path)
+        is_image = media_type and media_type.startswith("image")
+
         effective_nproc = 1 if is_image else self.NUM_GPUS_TOTAL
-        effective_spsize = 1 if is_image else spsize
-    
-        output_filename = f"result_{Path(filepath).stem}.mp4" if not is_image else f"{Path(filepath).stem}_upscaled"
-        output_filepath = outdir / output_filename
+        effective_sp_size = 1 if is_image else sp_size
 
 
+        output_filename = f"result_{Path(file_path).stem}.mp4"
+        output_filepath = out_dir / output_filename
+
         
         cmd = [
-            "torchrun", "--standalone", "--nnodes=1",
-            f"--nproc-per-node={effective_nproc}",
-            str(script),
-            "--video_path", str(filepath),
+            "torchrun", "--standalone", "--nnodes=1", f"--nproc-per-node={effective_nproc}", str(script),
+            "--video_path", str(file_path),
             "--output", str(output_filepath),
             "--model_dir", str(self.CKPTS_ROOT),
             "--seed", str(seed),
-            "--cuda_device", "0,1,2,3,4,5,6,7",
-            "--resolution", str(resh),
-            "--batch_size", str(effective_spsize),
+            "--cuda_device", str("0,1,2,3"),
+            "--resolution", str(res_h),
+            "--batch_size", str(effective_sp_size),
             "--model", "seedvr2_ema_3b_fp16.safetensors",
-            "--preserve_vram",
+            "--preserve_vram",                 
             "--debug",
-            "--output_format", "video" if not is_image else "png",
+            "--output_format", "video"
         ]
-    
-    
+
+        if effective_nproc > 1:
+            cmd = [c for c in cmd if not c.startswith('--cuda_device')]
+        
+        print("[SeedVRServer] Comando:", " ".join(cmd))
+        
+         
         print("SeedVRServer Comando:", " ".join(cmd))
         try:
             subprocess.run(cmd, cwd=str(self.SEEDVR_ROOT), check=True, env=os.environ.copy(), stdout=sys.stdout, stderr=sys.stderr)
@@ -109,3 +113,6 @@ class SeedVRServer:
         except Exception as e:
             print(f"[UI ERROR] A inferência falhou: {e}")
             return None, None, None
+
+
+