cleanup
This commit is contained in:
parent
9805635c12
commit
40249dd5ec
|
|
@ -465,7 +465,6 @@ class ModelBase:
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_model_architecture(cls, arch: str, model_type = ModelType.TEXT) -> type[ModelBase]:
|
def from_model_architecture(cls, arch: str, model_type = ModelType.TEXT) -> type[ModelBase]:
|
||||||
try:
|
try:
|
||||||
print(f"model_type: {model_type}, arch: {arch}")
|
|
||||||
return cls._model_classes[model_type][arch]
|
return cls._model_classes[model_type][arch]
|
||||||
except KeyError:
|
except KeyError:
|
||||||
raise NotImplementedError(f'Architecture {arch!r} not supported!') from None
|
raise NotImplementedError(f'Architecture {arch!r} not supported!') from None
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue