diff --git a/dorado/read_pipeline/BasecallerNode.cpp b/dorado/read_pipeline/BasecallerNode.cpp
index 008a1fc2..c39882f6 100644
--- a/dorado/read_pipeline/BasecallerNode.cpp
+++ b/dorado/read_pipeline/BasecallerNode.cpp
@@ -6,7 +6,6 @@
 #include "utils/stats.h"
 
 #include <nvtx3/nvtx3.hpp>
-#include <torch/nn.h>
 
 #include <algorithm>
 #include <cstdlib>