From 95db8d3ee6eb59481a7fc9793eab7e0971014f44 Mon Sep 17 00:00:00 2001
From: Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com>
Date: Wed, 11 Feb 2026 19:35:25 +0530
Subject: [PATCH] Fix SyncNet API call in train.py: get_vision_embed ->
 get_image_embed

The SyncNet model defines `get_image_embed()` in
musetalk/models/syncnet.py, but train.py incorrectly calls
`get_vision_embed()`, which does not exist. This causes an
AttributeError at runtime when `use_adapted_weight` is enabled.

The loss module (musetalk/loss/syncnet.py) already uses the correct
method name.
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 2c9f0ecd..e82dbf23 100755
--- a/train.py
+++ b/train.py
@@ -229,7 +229,7 @@ def get_ganloss_weight(step):
                     
                     # Calculate adapted weight based on audio-visual similarity
                     if cfg.use_adapted_weight:
-                        vision_embed_gt = syncnet.get_vision_embed(gt_frames)
+                        vision_embed_gt = syncnet.get_image_embed(gt_frames)
                         image_audio_sim_gt = F.cosine_similarity(
                             audio_embed, 
                             vision_embed_gt,