mend

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