Spaces:
Running
Running
Pedro Cuenca
commited on
Commit
·
5ec61cc
1
Parent(s):
1023afa
Use model configuration unless a specific one is supplied.
Browse files`from_pretrained` downloads the saved configuration if available, see
https://github.com/huggingface/transformers/blob/v4.15.0/src/transformers/modeling_utils.py#L1262.
- tools/train/train.py +1 -1
tools/train/train.py
CHANGED
|
@@ -458,7 +458,7 @@ def main():
|
|
| 458 |
if model_args.config_name:
|
| 459 |
config = DalleBartConfig.from_pretrained(model_args.config_name)
|
| 460 |
else:
|
| 461 |
-
config =
|
| 462 |
|
| 463 |
# Load or create new model
|
| 464 |
if model_args.model_name_or_path:
|
|
|
|
| 458 |
if model_args.config_name:
|
| 459 |
config = DalleBartConfig.from_pretrained(model_args.config_name)
|
| 460 |
else:
|
| 461 |
+
config = None
|
| 462 |
|
| 463 |
# Load or create new model
|
| 464 |
if model_args.model_name_or_path:
|