tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

feat: transformer fixed

Browse source
This commit is contained in:
Robin Meersman 2025-12-10 14:46:10 +01:00
parent f97c7c9130
commit d12bb25d0a
5 changed files with 65 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/models/__init__.py
View file

@ -1,9 +1,8 @@
from .Model import Model
from .cnn import CNNPredictor
from .transformer import Transformer
from .transformer import ByteTransformer
model_called: dict[str, type[Model]] = {
'cnn': CNNPredictor,
'transformer': Transformer
'transformer': ByteTransformer
}