Update src/facerender/modules/make_animation.py

src/facerender/modules/make_animation.py

@@ -162,22 +162,22 @@ def make_animation(source_image, source_semantics, target_semantics,
 
         with autocast():
             for frame_idx in tqdm(range(target_semantics.shape[1]), desc='Face Renderer:', unit='frame'):
-            target_semantics_frame = target_semantics[:, frame_idx]
-            he_driving = mapping(target_semantics_frame)
-
-            if yaw_c_seq is not None:
-                he_driving['yaw_in'] = yaw_c_seq[:, frame_idx]
-            if pitch_c_seq is not None:
-                he_driving['pitch_in'] = pitch_c_seq[:, frame_idx]
-            if roll_c_seq is not None:
-                he_driving['roll_in'] = roll_c_seq[:, frame_idx]
-
-            kp_driving = keypoint_transformation(kp_canonical, he_driving)
-            kp_norm = kp_driving
-            
-            out = generator(source_image, kp_source=kp_source, kp_driving=kp_norm)
-
-            predictions.append(out['prediction'])
+                target_semantics_frame = target_semantics[:, frame_idx]
+                he_driving = mapping(target_semantics_frame)
+    
+                if yaw_c_seq is not None:
+                    he_driving['yaw_in'] = yaw_c_seq[:, frame_idx]
+                if pitch_c_seq is not None:
+                    he_driving['pitch_in'] = pitch_c_seq[:, frame_idx]
+                if roll_c_seq is not None:
+                    he_driving['roll_in'] = roll_c_seq[:, frame_idx]
+    
+                kp_driving = keypoint_transformation(kp_canonical, he_driving)
+                kp_norm = kp_driving
+                
+                out = generator(source_image, kp_source=kp_source, kp_driving=kp_norm)
+    
+                predictions.append(out['prediction'])
             
         torch.cuda.synchronize()