Spaces:

mrowenyao
/

aptlm

Sleeping

mrowenyao commited on Jul 8

Commit

838a779

•

1 Parent(s): 8bc1979

cuda to cpu

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -17,7 +17,7 @@ def infer(prot, apta):
         n_heads=8,
         dropout=0.1,
         load_best_pt=True, # already loads the pretrained model using the datasets included in repo -- no need to run the bottom two cells
-        device='cuda',
         seed=1004)
     scores = pipeline.inference([apta], [prot], [0])

         n_heads=8,
         dropout=0.1,
         load_best_pt=True, # already loads the pretrained model using the datasets included in repo -- no need to run the bottom two cells
+        device='cpu',
         seed=1004)
     scores = pipeline.inference([apta], [prot], [0])