remove extra code

sahilg06 · Apr 9, 2023 · eb5eba1 · eb5eba1
1 parent f59e0ab
commit eb5eba1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/wav2lip.py b/models/wav2lip.py
@@ -115,7 +115,7 @@ def forward(self, audio_sequences, face_sequences, emotion):
 
  audio_embedding = self.audio_encoder(audio_sequences) # B*T, 512, 1, 1
  emotion_embedding = self.emotion_encoder(emotion)
- ee_needed = torch.mean(emotion_embedding,0).unsqueeze(0)
+ # ee_needed = torch.mean(emotion_embedding,0).unsqueeze(0)
 
  emotion_embedding = emotion_embedding.view(-1,512,1,1) # B*T, 512, 1, 1
 
@@ -147,7 +147,7 @@ def forward(self, audio_sequences, face_sequences, emotion):
  else:
  outputs = x
 
- return outputs,ee_needed
+ return outputs
 
 
 class Wav2Lip_disc_qual(nn.Module):