Spaces:

flax-community
/

dalle-mini

Running

Pedro Cuenca commited on Jan 16, 2022

Commit

290e443

•

1 Parent(s): bb3f53e

Update help string for `model_name_or_path`.

Files changed (1) hide show

tools/train/train.py CHANGED Viewed

@@ -58,8 +58,9 @@ class ModelArguments:
     model_name_or_path: Optional[str] = field(
         default=None,
         metadata={
-            "help": "The model checkpoint for weights initialization."
-            "Don't set if you want to train a model from scratch."
         },
     )
     config_name: Optional[str] = field(

     model_name_or_path: Optional[str] = field(
         default=None,
         metadata={
+            "help": "The model checkpoint for weights initialization. "
+            "Don't set if you want to train a model from scratch. "
+            "W&B artifact references are supported in addition to the sources supported by `PreTrainedModel`."
         },
     )
     config_name: Optional[str] = field(