diff --git a/main.nf b/main.nf
index a091e31..fd15497 100644
--- a/main.nf
+++ b/main.nf
@@ -54,7 +54,7 @@ process FLOWDOCK {
             visualize_sample_trajectories=false \\
             auxiliary_estimation_only=false \\
             esmfold_chunk_size=null \\
-            trainer=gpu 2>&1 | tee run.log
+            trainer=cpu 2>&1 | tee run.log
         
         # Copy any outputs from FlowDock's log directory if they exist there
         if ls /software/flowdock/logs/sample/runs/*/rank*.pdb 2>/dev/null; then