Merge pull request #339 from mims-harvard/scgpt

use tdc geneformer
mims-harvard · Jan 7, 2025 · c0de53e · c0de53e
2 parents 78f457e + e7864fc
commit c0de53e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tdc/model_server/tdc_hf.py b/tdc/model_server/tdc_hf.py
@@ -57,8 +57,7 @@ def load(self):
             raise Exception("this model is not in the TDC model hub GH repo.")
         elif self.model_name == "Geneformer":
             from transformers import AutoModelForMaskedLM
-            model = AutoModelForMaskedLM.from_pretrained(
-                "ctheodoris/Geneformer")
+            model = AutoModelForMaskedLM.from_pretrained("tdc/Geneformer")
             return model
         elif self.model_name == "scGPT":
             from transformers import AutoModel