Update pipe.py
Browse files
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):
|