Spaces:

ChemFM
/

molecular_conditional_generation

Running on Zero

feiyang-cai commited on Jan 25, 2025

Commit

5c66cc6

1 Parent(s): 1106177

update

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -256,10 +256,10 @@ class MolecularGenerationModel():
             has_scaffold=False
         )
-    @spaces.GPU(duration=20)
     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):

             has_scaffold=False
         )
+    #@spaces.GPU(duration=20)
     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):