From 9bcd74881754df1e0e89f4aa098153e745b4bc8e Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Thu, 26 Mar 2026 20:37:12 +0800
Subject: [PATCH] Update test_wanx_tia2mv_obj_back.py

---
 test_wanx_tia2mv_obj_back.py | 218 ++++++++++++++++++-----------------
 1 file changed, 111 insertions(+), 107 deletions(-)

diff --git a/test_wanx_tia2mv_obj_back.py b/test_wanx_tia2mv_obj_back.py
index b556b6d..f5d6505 100644
--- a/test_wanx_tia2mv_obj_back.py
+++ b/test_wanx_tia2mv_obj_back.py
@@ -21,119 +21,123 @@ from .wan.configs import WAN_CONFIGS
 from .utils.img_utils import process_images_final,resize_images,resize_short_side
 from .model_loader_utils import covert_obj_img,clear_comfyui_cache
 
-mean = torch.tensor(
-            [
-                -0.2289,
-                -0.0052,
-                -0.1323,
-                -0.2339,
-                -0.2799,
-                0.0174,
-                0.1838,
-                0.1557,
-                -0.1382,
-                0.0542,
-                0.2813,
-                0.0891,
-                0.1570,
-                -0.0098,
-                0.0375,
-                -0.1825,
-                -0.2246,
-                -0.1207,
-                -0.0698,
-                0.5109,
-                0.2665,
-                -0.2108,
-                -0.2158,
-                0.2502,
-                -0.2055,
-                -0.0322,
-                0.1109,
-                0.1567,
-                -0.0729,
-                0.0899,
-                -0.2799,
-                -0.1230,
-                -0.0313,
-                -0.1649,
-                0.0117,
-                0.0723,
-                -0.2839,
-                -0.2083,
-                -0.0520,
-                0.3748,
-                0.0152,
-                0.1957,
-                0.1433,
-                -0.2944,
-                0.3573,
-                -0.0548,
-                -0.1681,
-                -0.0667,
-            ],
-            device=torch.device('cuda'),
-        )
-std = torch.tensor(
-            [
-                0.4765,
-                1.0364,
-                0.4514,
-                1.1677,
-                0.5313,
-                0.4990,
-                0.4818,
-                0.5013,
-                0.8158,
-                1.0344,
-                0.5894,
-                1.0901,
-                0.6885,
-                0.6165,
-                0.8454,
-                0.4978,
-                0.5759,
-                0.3523,
-                0.7135,
-                0.6804,
-                0.5833,
-                1.4146,
-                0.8986,
-                0.5659,
-                0.7069,
-                0.5338,
-                0.4889,
-                0.4917,
-                0.4069,
-                0.4999,
-                0.6866,
-                0.4093,
-                0.5709,
-                0.6065,
-                0.6415,
-                0.4944,
-                0.5726,
-                1.2042,
-                0.5458,
-                1.6887,
-                0.3971,
-                1.0600,
-                0.3943,
-                0.5537,
-                0.5444,
-                0.4089,
-                0.7468,
-                0.7744,
-            ],
-            device=torch.device('cuda'),
-        )
-wanvae_scale = [mean, 1.0 / std]
+def get_mu_scale():
+    mean_InteractAvatar = torch.tensor(
+                [
+                    -0.2289,
+                    -0.0052,
+                    -0.1323,
+                    -0.2339,
+                    -0.2799,
+                    0.0174,
+                    0.1838,
+                    0.1557,
+                    -0.1382,
+                    0.0542,
+                    0.2813,
+                    0.0891,
+                    0.1570,
+                    -0.0098,
+                    0.0375,
+                    -0.1825,
+                    -0.2246,
+                    -0.1207,
+                    -0.0698,
+                    0.5109,
+                    0.2665,
+                    -0.2108,
+                    -0.2158,
+                    0.2502,
+                    -0.2055,
+                    -0.0322,
+                    0.1109,
+                    0.1567,
+                    -0.0729,
+                    0.0899,
+                    -0.2799,
+                    -0.1230,
+                    -0.0313,
+                    -0.1649,
+                    0.0117,
+                    0.0723,
+                    -0.2839,
+                    -0.2083,
+                    -0.0520,
+                    0.3748,
+                    0.0152,
+                    0.1957,
+                    0.1433,
+                    -0.2944,
+                    0.3573,
+                    -0.0548,
+                    -0.1681,
+                    -0.0667,
+                ],
+                device=torch.device('cuda'),
+            )
+    std_InteractAvatar = torch.tensor(
+                [
+                    0.4765,
+                    1.0364,
+                    0.4514,
+                    1.1677,
+                    0.5313,
+                    0.4990,
+                    0.4818,
+                    0.5013,
+                    0.8158,
+                    1.0344,
+                    0.5894,
+                    1.0901,
+                    0.6885,
+                    0.6165,
+                    0.8454,
+                    0.4978,
+                    0.5759,
+                    0.3523,
+                    0.7135,
+                    0.6804,
+                    0.5833,
+                    1.4146,
+                    0.8986,
+                    0.5659,
+                    0.7069,
+                    0.5338,
+                    0.4889,
+                    0.4917,
+                    0.4069,
+                    0.4999,
+                    0.6866,
+                    0.4093,
+                    0.5709,
+                    0.6065,
+                    0.6415,
+                    0.4944,
+                    0.5726,
+                    1.2042,
+                    0.5458,
+                    1.6887,
+                    0.3971,
+                    1.0600,
+                    0.3943,
+                    0.5537,
+                    0.5444,
+                    0.4089,
+                    0.7468,
+                    0.7744,
+                ],
+                device=torch.device('cuda'),
+            )
+    wanvae_scale = [mean_InteractAvatar, 1.0 / std_InteractAvatar]
+    return wanvae_scale
 
 def get_mu_scale(mu):
+    wanvae_scale = get_mu_scale()
     mu = (mu - wanvae_scale[0].view(1, 48, 1, 1, 1).to(mu.device,mu.dtype)) * wanvae_scale[1].view(
                         1, 48, 1, 1, 1).to(mu.device,mu.dtype)
     return mu
 def get_z_scale(z):
+    wanvae_scale=get_mu_scale()
     z = z / wanvae_scale[1].view(1, 48, 1, 1, 1).to(z.device,z.dtype) + wanvae_scale[0].view(
                     1, 48, 1, 1, 1).to(z.device,z.dtype)
     return z