From 305a52dde8d3e43437e453ff5b634334b49d1f1c Mon Sep 17 00:00:00 2001 From: Juarez Bochi Date: Mon, 18 Dec 2023 11:25:14 -0500 Subject: [PATCH] Run hf_t5 with any model --- t5/hf_t5.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/t5/hf_t5.py b/t5/hf_t5.py index cb75c0f6..da269e2e 100644 --- a/t5/hf_t5.py +++ b/t5/hf_t5.py @@ -28,9 +28,6 @@ if __name__ == "__main__": ) parser.add_argument( "--model", - choices=[ - "t5-small", - ], default="t5-small", help="The huggingface name of the T5 model to save.", )