Update peptide/moo.py

peptide/moo.py

@@ -94,7 +94,7 @@ class MOGGenerator:
     def generate(self):
         """Main generation loop."""
         shape = (self.args.num_samples, self.args.gen_len + 2)
-        x = torch.randint(5, self.model.vocab_size, shape, dtype=torch.long, device=self.device)
+        x = torch.randint(4, self.model.vocab_size, shape, dtype=torch.long, device=self.device)
         x[:, 0] = 0
         x[:, -1] = 2