diff --git a/infer.py b/infer.py
index b15f94d..39bffce 100644
--- a/infer.py
+++ b/infer.py
@@ -25,7 +25,6 @@ def main(args):
         textual_transformer=True,
         load_trained=args.trained_gfun,
         load_meta=True,
-        trained_text_trf=args.trained_transformer,
     )
 
     predictions = gfun.transform(lX)
@@ -42,7 +41,6 @@ if __name__ == "__main__":
     parser.add_argument("--nlabels", type=int, default=28)
     parser.add_argument("--muse_dir", type=str, default="~/resources/muse_embeddings", help="Path to muse embeddings")
     parser.add_argument("--trained_gfun", type=str, default="rai_pmt_mean_231029", help="name of the trained gfun instance")
-    parser.add_argument("--trained_transformer", default="hf_models/mbert-fewshot-rai-full/checkpoint-5150-small", help="path to fine-tuned transformer")
     args = parser.parse_args()
     main(args)