Run hf_t5 with any model

This commit is contained in:
Juarez Bochi 2023-12-18 11:25:14 -05:00
parent 0779417903
commit 305a52dde8
No known key found for this signature in database
GPG Key ID: 34CCBB77DC8BEBB6

View File

@ -28,9 +28,6 @@ if __name__ == "__main__":
) )
parser.add_argument( parser.add_argument(
"--model", "--model",
choices=[
"t5-small",
],
default="t5-small", default="t5-small",
help="The huggingface name of the T5 model to save.", help="The huggingface name of the T5 model to save.",
) )