From 639425ad7d8b501cbe7d34f27db4c0676d8e22a6 Mon Sep 17 00:00:00 2001 From: faraphel Date: Fri, 10 Jan 2025 00:25:07 +0100 Subject: [PATCH] fixed model information being incorrect --- source/model/base/BaseModel.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/source/model/base/BaseModel.py b/source/model/base/BaseModel.py index ce763b0..d94dfca 100644 --- a/source/model/base/BaseModel.py +++ b/source/model/base/BaseModel.py @@ -3,7 +3,6 @@ import gc import typing from pathlib import Path -from source import api from source.manager import ModelManager @@ -42,7 +41,7 @@ class BaseModel(abc.ABC): return { "name": self.name, - "response_mimetype": self.response_mimetype, + "output_type": self.output_type, } def load(self) -> None: