Update ESM1b model ID format to use underscores

VishwamAI · Nov 14, 2024 · 84ae01e · 84ae01e
1 parent 60c0934
commit 84ae01e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/analysis/enhanced_sequence_analyzer.py b/models/analysis/enhanced_sequence_analyzer.py
@@ -18,8 +18,8 @@ def __init__(self, config: Dict):
         self.hidden_size = config.get('hidden_size', 768)
 
         # Initialize protein language model
-        self.tokenizer = AutoTokenizer.from_pretrained('facebook/esm1b-t24-1M')
-        self.protein_model = AutoModel.from_pretrained('facebook/esm1b-t24-1M')
+        self.tokenizer = AutoTokenizer.from_pretrained('facebook/esm1b_t24_1M')
+        self.protein_model = AutoModel.from_pretrained('facebook/esm1b_t24_1M')
 
         # Feature extraction layers
         self.feature_extractor = nn.Sequential(