Spaces:

ChemFM
/

molecular_conditional_generation

Running on Zero

feiyang-cai commited on Jan 25, 2025

Commit

283ddb7

verified ·

1 Parent(s): 6967f07

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -256,7 +256,7 @@ class MolecularGenerationModel():
             has_scaffold=False
         )
-    @spaces.GPU(duration=60)
     def generate(self, loader):
         self.model.to("cuda")
         self.model.eval()

             has_scaffold=False
         )
+    @spaces.GPU(duration=20)
     def generate(self, loader):
         self.model.to("cuda")
         self.model.eval()