From 8cb4ac3cad41ceda67a338e63df97745d78bbbd2 Mon Sep 17 00:00:00 2001
From: Isi <isinsemail@gmail.com>
Date: Tue, 11 Mar 2025 11:07:56 +0100
Subject: [PATCH] Update inference_animate_x_entrance.py

---
 tools/inferences/inference_animate_x_entrance.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/inferences/inference_animate_x_entrance.py b/tools/inferences/inference_animate_x_entrance.py
index ca456b5..6cd2f91 100644
--- a/tools/inferences/inference_animate_x_entrance.py
+++ b/tools/inferences/inference_animate_x_entrance.py
@@ -70,11 +70,11 @@ def inference_animate_x_entrance(seed, steps, useFirstFrame, reference_image, re
         cfg.gpus_per_machine = torch.cuda.device_count()
         cfg.world_size = cfg.pmi_world_size * cfg.gpus_per_machine
     
-    if cfg.world_size == 1:
+    if cfg.world_size >= 1:
         return worker(0, seed, steps, useFirstFrame, reference_image, ref_pose, pose_sequence, frame_interval, max_frames, resolution, cfg, cfg_update)
         # return worker(0, seed, steps, useFirstFrame, reference_image, ref_pose, pose_sequence, original_driven_video_path, refpose_embeding_key, pose_embedding_key,frame_interval, max_frames, resolution, cfg, cfg_update)
-    else:
-        return mp.spawn(worker, nprocs=cfg.gpus_per_machine, args=(cfg, cfg_update))
+    # else:
+    #     return mp.spawn(worker, nprocs=cfg.gpus_per_machine, args=(cfg, cfg_update))
     return cfg