Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated inference
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 9m34s Details

This commit is contained in:
Falko Victor Habel 2025-07-02 22:21:35 +02:00
parent 5d0aadebab
commit ff45f54860
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/aiunn/inference/inference.py
View File

@ -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()