justheuristic il y a 3 ans
Parent
commit
04797c55e4
1 fichiers modifiés avec 1 ajouts et 2 suppressions
  1. 1 2
      cli/convert_model.py

+ 1 - 2
cli/convert_model.py

@@ -35,8 +35,7 @@ if __name__ == "__main__":
         "--commit_message", type=str, default="push-o-matic", help="Use this commit message for all parts"
     )
     parser.add_argument("--use_auth_token", type=str, default=None, help="auth token for from_pretrained")
-    parser.add_argument("--resize_token_embeddings", type=int, default=None,
-                        help="change the vocabulary size of the converted model to this value")
+    parser.add_argument("--resize_token_embeddings", type=int, default=None, help="change the vocabulary size")
     args = parser.parse_args()
 
     free_ram_gb = psutil.virtual_memory().available / 2**30