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.
able2608 · Aug 8, 2024 · ba86f12 · ba86f12
1 parent 83b1cdb
commit ba86f12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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']