Merge pull request #122 from LWprogramming/eos_id

Use correct eos id when masking out

Author: lucidrains

audiolm_pytorch/audiolm_pytorch.py

@@ -1135,7 +1135,7 @@ def generate(
 
             last_logit_indices += 1
 
-        sample_semantic_ids = mask_out_after_eos_id(sample_semantic_ids, self.pad_id, keep_eos = False)
+        sample_semantic_ids = mask_out_after_eos_id(sample_semantic_ids, self.eos_id, keep_eos = False)
 
         return sample_semantic_ids