8 lines
200 B
Python
8 lines
200 B
Python
from .Model import Model
|
|
from .cnn import CNNPredictor
|
|
from .transformer import ByteTransformer
|
|
|
|
model_called: dict[str, type[Model]] = {
|
|
'cnn': CNNPredictor,
|
|
'transformer': ByteTransformer
|
|
}
|