alegendaryfish commited on
Commit
301a25c
·
verified ·
1 Parent(s): 6ef331a

Rename internal model references to CodonTranslatorModel

Browse files
Files changed (1) hide show
  1. src/models.py +2 -2
src/models.py CHANGED
@@ -1,6 +1,6 @@
1
  """
2
  Core model architectures for CodonTranslator.
3
- - Decoder-only backbone with species + protein prefixing
4
  Includes a frozen ESM-C encoder for protein conditioning.
5
  """
6
 
@@ -97,7 +97,7 @@ class FrozenESMCEncoder(nn.Module):
97
 
98
 
99
 
100
- class CodonGPT(nn.Module):
101
  def __init__(
102
  self,
103
  vocab_size: int = 79,
 
1
  """
2
  Core model architectures for CodonTranslator.
3
+ - CodonTranslatorModel: decoder-only backbone with species + protein prefixing
4
  Includes a frozen ESM-C encoder for protein conditioning.
5
  """
6
 
 
97
 
98
 
99
 
100
+ class CodonTranslatorModel(nn.Module):
101
  def __init__(
102
  self,
103
  vocab_size: int = 79,