Spaces:

ChemFM
/

molecular_conditional_generation

Running on Zero

feiyang-cai commited on Jan 25

Commit

21ff76d

1 Parent(s): a18ead3

update

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -258,7 +258,8 @@ class MolecularGenerationModel():
     @spaces.GPU(duration=60)
     def generate(self, loader):
         df = []
         pbar = tqdm(loader, desc=f"Evaluating...", leave=False)
         for it, batch in enumerate(pbar):

     @spaces.GPU(duration=60)
     def generate(self, loader):
+        self.model.to("cuda")
+        self.model.eval()
         df = []
         pbar = tqdm(loader, desc=f"Evaluating...", leave=False)
         for it, batch in enumerate(pbar):