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