Update tdc_hf.py

mims-harvard · Jan 7, 2025 · a684dd3 · a684dd3
1 parent e6268fb
commit a684dd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tdc/model_server/tdc_hf.py b/tdc/model_server/tdc_hf.py
@@ -57,7 +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("tdc/Geneformer")
+            model = AutoModelForMaskedLM.from_pretrained("ctheodoris/Geneformer")
             return model
         elif self.model_name == "scGPT":
             from transformers import AutoModel