Update FantasyTalking/model.py

FantasyTalking/model.py

@@ -97,10 +97,8 @@ class FantasyTalkingAudioConditionModel(nn.Module):
 
         # audio proj model
         self.proj_model = self.init_proj(self.audio_proj_dim)
-        #self.set_audio_processor(wan_dit)
-        #هذا الجيد
-        if wan_dit is not None:
-            self.set_audio_processor(wan_dit)
+        self.set_audio_processor(wan_dit)
+      
 
     def init_proj(self, cross_attention_dim=5120):
         proj_model = AudioProjModel(
@@ -132,9 +130,7 @@ class FantasyTalkingAudioConditionModel(nn.Module):
         else:
             state_dict = torch.load(ip_ckpt, map_location="cpu")
         self.proj_model.load_state_dict(state_dict["proj_model"])
-        #wan_dit.load_state_dict(state_dict["audio_processor"], strict=False)
-        if wan_dit is not None and "audio_processor" in state_dict:
-            wan_dit.load_state_dict(state_dict["audio_processor"], strict=False)
+        wan_dit.load_state_dict(state_dict["audio_processor"], strict=False)
 
 
     def get_proj_fea(self, audio_fea=None):