Update pipe.py

pipe.py

@@ -79,7 +79,7 @@ class SmilesDiffusionPipe(Pipeline):
         return torch.multinomial(probs, 1)
 
 
-    def unmask_smiles_argmax(self, input_ids, model, voc):
+    def unmask_partial_smiles(self, input_ids, model, voc):
         """
         Unmasks all [MASK] tokens in a single forward pass using argmax decoding.
         """