diff --git a/src/live_portrait_wrapper.py b/src/live_portrait_wrapper.py
index d2a678d..c7f65c9 100644
--- a/src/live_portrait_wrapper.py
+++ b/src/live_portrait_wrapper.py
@@ -52,9 +52,9 @@ class LivePortraitWrapper(object):
         # Optimize for inference
         if self.compile:
             torch._dynamo.config.suppress_errors = True  # Suppress errors and fall back to eager execution
-            self.warping_module = torch.compile(self.warping_module, mode='max-autotune')  
-            self.spade_generator = torch.compile(self.spade_generator, mode='max-autotune')  
-        
+            self.warping_module = torch.compile(self.warping_module, mode='max-autotune')
+            self.spade_generator = torch.compile(self.spade_generator, mode='max-autotune')
+
         self.timer = Timer()
 
     def update_config(self, user_args):
@@ -199,7 +199,7 @@ class LivePortraitWrapper(object):
         """
         kp_source: BxNx3
         eye_close_ratio: Bx3
-        Return: Bx(3*num_kp+2)
+        Return: Bx(3*num_kp)
         """
         feat_eye = concat_feat(kp_source, eye_close_ratio)
 
@@ -212,6 +212,7 @@ class LivePortraitWrapper(object):
         """
         kp_source: BxNx3
         lip_close_ratio: Bx2
+        Return: Bx(3*num_kp)
         """
         feat_lip = concat_feat(kp_source, lip_close_ratio)