Update skyreelsinfer/skyreels_video_infer.py

skyreelsinfer/skyreels_video_infer.py

@@ -81,18 +81,10 @@ class SkyReelsVideoSingleGpuInfer:
     ):
         self.task_type = task_type
         self.gpu_rank = local_rank
-        dist.init_process_group(
-            backend="nccl",
-            init_method="tcp://127.0.0.1:23456",
-            timeout=timedelta(seconds=600),
-            world_size=world_size,
-            rank=local_rank,
-        )
         os.environ["LOCAL_RANK"] = str(local_rank)
-        logger.info(f"rank:{local_rank} Distributed backend: {dist.get_backend()}")
-        torch.cuda.set_device(dist.get_rank())
+        torch.cuda.set_device(0)
         torch.backends.cuda.enable_cudnn_sdp(False)
-        gpu_device = f"cuda:{dist.get_rank()}"
+        gpu_device = "cuda:0"
 
         self.pipe: SkyreelsVideoPipeline = self._load_model(
             model_id=model_id, quant_model=quant_model, gpu_device=gpu_device