From ba86f124b435d37cd14193409846c786f5947497 Mon Sep 17 00:00:00 2001
From: Isi <86603298+Isi-dev@users.noreply.github.com>
Date: Thu, 8 Aug 2024 14:31:19 +0100
Subject: [PATCH] Update inference_unianimate_entrance.py

Removed the space on line 284 at the end of unifiedModel = os.path.join(root_directory, 'checkpoints/unianimate_16f_32f_non_ema_223000.pth ') which seems to prevent loading of the main model in some environments.
---
 tools/inferences/inference_unianimate_entrance.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/inferences/inference_unianimate_entrance.py b/tools/inferences/inference_unianimate_entrance.py
index 851c549..df18f72 100644
--- a/tools/inferences/inference_unianimate_entrance.py
+++ b/tools/inferences/inference_unianimate_entrance.py
@@ -281,7 +281,7 @@ def worker(gpu, seed, steps, useFirstFrame, reference_image, ref_pose, pose_sequ
     parent_directory = os.path.dirname(current_directory)
     # uniAnimate folder
     root_directory = os.path.dirname(parent_directory)
-    unifiedModel = os.path.join(root_directory, 'checkpoints/unianimate_16f_32f_non_ema_223000.pth ')
+    unifiedModel = os.path.join(root_directory, 'checkpoints/unianimate_16f_32f_non_ema_223000.pth')
     state_dict = torch.load(unifiedModel, map_location='cpu')
     if 'state_dict' in state_dict:
         state_dict = state_dict['state_dict']