diff --git a/atoma-inference/src/lib.rs b/atoma-inference/src/lib.rs
index 80915305..d31feaad 100644
--- a/atoma-inference/src/lib.rs
+++ b/atoma-inference/src/lib.rs
@@ -1,5 +1,7 @@
 pub mod apis;
 pub mod candle;
+pub mod config;
+pub mod core_thread;
 pub mod model_thread;
 pub mod models;
 pub mod service;