Test2

Paused

EuuIia commited on Oct 4

Commit

fc974f1

verified ·

1 Parent(s): 7b75d8d

Update api/seedvr_server.py

Files changed (1) hide show

api/seedvr_server.py CHANGED Viewed

@@ -85,7 +85,7 @@ class SeedVRServer:
             "--output", str(output_filepath),
             "--model_dir", str(self.CKPTS_ROOT),
             "--seed", str(seed),
-            "--cuda_device", "0",
             "--resolution", str(resh),
             "--batch_size", str(effective_spsize),
             "--model", "seedvr2_ema_3b_fp16.safetensors",

             "--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),
             "--model", "seedvr2_ema_3b_fp16.safetensors",