back to tdc geneformer

mims-harvard · Jan 7, 2025 · b118fc7 · b118fc7
1 parent e879f11
commit b118fc7
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