From ff45f54860bf71aedb5f4cbaff0081de1d47681c Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Wed, 2 Jul 2025 22:21:35 +0200 Subject: [PATCH] updated inference --- src/aiunn/inference/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/aiunn/inference/inference.py b/src/aiunn/inference/inference.py index 1ff5994..953cc0b 100644 --- a/src/aiunn/inference/inference.py +++ b/src/aiunn/inference/inference.py @@ -30,7 +30,7 @@ class aiuNNInference: self.device = device # Load the model with specified precision - self.model = aiuNN.from_pretrained(model_path) + self.model = aiuNN.from_pretrained(model_path,) self.model.to(self.device) self.model.eval()