From 864c892e7fc6993b89b548c51c42379a50c26922 Mon Sep 17 00:00:00 2001
From: Martin Stefcek <gcifko@gmail.com>
Date: Thu, 4 Apr 2024 17:50:58 +0400
Subject: [PATCH] fix

---
 atoma-inference/src/service.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/atoma-inference/src/service.rs b/atoma-inference/src/service.rs
index 33a88b34..d3e08830 100644
--- a/atoma-inference/src/service.rs
+++ b/atoma-inference/src/service.rs
@@ -213,7 +213,11 @@ mod tests {
             Ok(())
         }
 
-        fn load(_: Vec<PathBuf>, _: PrecisionBits) -> Result<Self, crate::models::ModelError> {
+        fn load(
+            _: Vec<PathBuf>,
+            _: PrecisionBits,
+            _device_id: usize,
+        ) -> Result<Self, crate::models::ModelError> {
             Ok(Self {})
         }