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.", )