From b118fc77e50850b06403b32027f7ffc7941e91f8 Mon Sep 17 00:00:00 2001 From: Alejandro Velez Date: Tue, 7 Jan 2025 06:44:11 -0500 Subject: [PATCH] back to tdc geneformer --- tdc/model_server/tdc_hf.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tdc/model_server/tdc_hf.py b/tdc/model_server/tdc_hf.py index 223012d1..57e25473 100644 --- 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