diff --git a/.gitignore b/.gitignore
index 907f3fd..db6d490 100644
--- a/.gitignore
+++ b/.gitignore
@@ -593,4 +593,5 @@ scripts/*.json
 *stats_*.json
 !build.rs
 .ruff_cache
-hg_deepchopper
\ No newline at end of file
+hg_deepchopper
+analysis_data
\ No newline at end of file
diff --git a/deepchopper/cli.py b/deepchopper/cli.py
index f1c77ef..8a10fe4 100644
--- a/deepchopper/cli.py
+++ b/deepchopper/cli.py
@@ -135,7 +135,7 @@ def predict(
 
     trainer = lightning.pytorch.trainer.Trainer(
         accelerator=accelerator,
-        devices=gpus,
+        devices=devices,
         callbacks=callbacks,
         deterministic=False,
         logger=False,