NOTE: if this is not a bug report, please use the GitHub Discussions for support questions (How do I do X?), feature requests, ideas, showcasing new applications, etc.
Bug descriptionGVPTransformerModel.sample samples from all tokens including \<eos>, \<mask>, etc.
Reproduction steps
Run the example inverse folding script for sequence sampling with a high temperature. I've seen these tokens appear with a temperature as low as 3, but try 10.
Expected behavior
I expect the tokens sampled to be only for amino acids since only amino acids make sense as outputs for structure residues.
NOTE: if this is not a bug report, please use the GitHub Discussions for support questions (How do I do X?), feature requests, ideas, showcasing new applications, etc.
Bug description
GVPTransformerModel.sample
samples from all tokens including \<eos>, \<mask>, etc.Reproduction steps Run the example inverse folding script for sequence sampling with a high temperature. I've seen these tokens appear with a temperature as low as 3, but try 10.
Expected behavior I expect the tokens sampled to be only for amino acids since only amino acids make sense as outputs for structure residues.