Failed to Run: Swap to Many Faces

I have tried to swap multi faces, here is the issue I have encountered. Please share thoughts, if anyone is able to replicate the issue or has fixed it.

```
python3 inference.py --source_paths multispecific/SRC_01.png multispecific/SRC_02.png multispecific/SRC_03.png --target_faces_paths multispecific/DST_01.jpg multispecific/DST_02.jpg multispecific/DST_03.jpg --target_videomultispecific/multi_3s.mp4 --out_video_name out.mp4

```
```
Traceback (most recent call last):
File "inference.py", line 153, in
main(args)
File "inference.py", line 104, in main
BS=args.batch_size)
File "/content/drive/MyDrive/sber-swap/utils/inference/core.py", line 63, in model_inference
target_batch_rs = transform_target_to_torch(resized_frs, half=half)
File "/content/drive/MyDrive/sber-swap/utils/inference/core.py", line 18, in transform_target_to_torch
target_batch_rs = target_batch_rs[:, :, :, [2,1,0]]/255.
IndexError: too many indices for tensor of dimension 1
```