Rename internal model references to CodonTranslatorModel

train.py

@@ -13,7 +13,7 @@ import argparse
 import logging
 import torch
 
-from src import CodonGPT, CodonTokenizer, Trainer, TrainingArguments
+from src import CodonTranslatorModel, CodonTokenizer, Trainer, TrainingArguments
 from src.dataset import create_precomputed_dataloaders, SpeciesEmbeddingStore
 
 logging.basicConfig(
@@ -241,7 +241,7 @@ def main():
         esm_dev = f"cuda:{lr}"
 
     # Model — species is always on; protein defaults to ON (can be disabled with --no_protein)
-    model = CodonGPT(
+    model = CodonTranslatorModel(
         vocab_size=tok.vocab_size,
         num_special_tokens=tok.num_special_tokens,
         special_ids=tok.special_ids,