fix

atoma-network · Apr 4, 2024 · 864c892 · 864c892
1 parent bddd088
commit 864c892
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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 {})
         }