Rename internal model references to CodonTranslatorModel
Browse files- src/models.py +2 -2
src/models.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
"""
|
| 2 |
Core model architectures for CodonTranslator.
|
| 3 |
-
-
|
| 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
|
| 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,
|