kelu01 commited on
Commit
36d7471
·
verified ·
1 Parent(s): 62fd606

Update pipe.py

Browse files
Files changed (1) hide show
  1. pipe.py +0 -7
pipe.py CHANGED
@@ -41,13 +41,6 @@ class SmilesDiffusionPipe(Pipeline):
41
  def preprocess(self, smiles_str, steps=30, k=1):
42
  tokenized = self.tokenize_smiles(smiles_str)
43
  encoded = self.voc.encode(tokenized, is_split_into_words=True)
44
-
45
- # ✅ DEBUG PRINTS
46
- print("Original SMILES:", smiles_str)
47
- print("Tokenized SMILES:", tokenized)
48
- print("Encoded input_ids:", encoded["input_ids"])
49
- print("Tokens from IDs:", self.voc.convert_ids_to_tokens(encoded["input_ids"]))
50
-
51
  return (encoded, steps, k)
52
 
53
  def _forward(self, inputs):
 
41
  def preprocess(self, smiles_str, steps=30, k=1):
42
  tokenized = self.tokenize_smiles(smiles_str)
43
  encoded = self.voc.encode(tokenized, is_split_into_words=True)
 
 
 
 
 
 
 
44
  return (encoded, steps, k)
45
 
46
  def _forward(self, inputs):