nodes/smthemex-ComfyUI_InteractAvatar
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update test_wanx_tia2mv_obj_back.py

Browse Source
This commit is contained in:
smthemex
2026-06-03 17:50:10 +08:00
committed by GitHub
parent 9bcd748817
commit 9964212c57
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_wanx_tia2mv_obj_back.py
+3 -3
View File
@@ -21,7 +21,7 @@ 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
def get_mu_scale():
def get_mu_scale_():
mean_InteractAvatar = torch.tensor(
[
-0.2289,
@@ -132,12 +132,12 @@ def get_mu_scale():
return wanvae_scale
def get_mu_scale(mu):
wanvae_scale = get_mu_scale()
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()
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