Spaces:

ChemFM
/

molecular_conditional_generation

Sleeping

feiyang-cai commited on Jan 25

Commit

6500e94

verified ·

1 Parent(s): 3d13639

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -283,7 +283,6 @@ class MolecularGenerationModel():
             input_length = batch['input_ids'].shape[1]
             steps = 1024 - input_length
-            print(self.model.device, "model_device")
             with torch.set_grad_enabled(False):
                 early_stop_flags = torch.zeros(num_generations, dtype=torch.bool).to(self.model.device)
                 for k in range(steps):

             input_length = batch['input_ids'].shape[1]
             steps = 1024 - input_length
             with torch.set_grad_enabled(False):
                 early_stop_flags = torch.zeros(num_generations, dtype=torch.bool).to(self.model.device)
                 for k in range(steps):