fixed model information being incorrect
This commit is contained in:
parent
7bd84c8570
commit
639425ad7d
1 changed files with 1 additions and 2 deletions
|
@ -3,7 +3,6 @@ import gc
|
||||||
import typing
|
import typing
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from source import api
|
|
||||||
from source.manager import ModelManager
|
from source.manager import ModelManager
|
||||||
|
|
||||||
|
|
||||||
|
@ -42,7 +41,7 @@ class BaseModel(abc.ABC):
|
||||||
|
|
||||||
return {
|
return {
|
||||||
"name": self.name,
|
"name": self.name,
|
||||||
"response_mimetype": self.response_mimetype,
|
"output_type": self.output_type,
|
||||||
}
|
}
|
||||||
|
|
||||||
def load(self) -> None:
|
def load(self) -> None:
|
||||||
|
|
Loading…
Reference in a new issue