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