Spaces:

yangliz5
/

deepchopper

Running

yangliz5 commited on Oct 11, 2024

Commit

be3fb72

1 Parent(s): 2c11490

refactor: Remove unnecessary device selection in predict function

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,6 @@ def predict(
     accelerator = "cpu" if torch.cuda.is_available() else "gpu"
     trainer = lightning.pytorch.trainer.Trainer(
         accelerator=accelerator,
-        devices=-1,
         deterministic=False,
         logger=False,
     )

     accelerator = "cpu" if torch.cuda.is_available() else "gpu"
     trainer = lightning.pytorch.trainer.Trainer(
         accelerator=accelerator,
         deterministic=False,
         logger=False,
     )