nodes/Isi-dev-ComfyUI-UniAnimate-W
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update inference_animate_x_entrance.py

Browse Source
This commit is contained in:
Isi
2025-03-11 11:07:56 +01:00
committed by GitHub
parent 374a1b0c85
commit 8cb4ac3cad
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/inferences/inference_animate_x_entrance.py
+3 -3
View File
@@ -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