Skip to content

Commit caf4890

Browse files
Update preset map for T5 model (#2414)
1 parent 6df4820 commit caf4890

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tools/checkpoint_conversion/convert_t5_checkpoints.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,9 @@
1111
import keras_hub
1212

1313
PRESET_MAP = {
14-
"t5_small_multi": "t5-small",
15-
"t5_base_multi": "t5-base",
16-
"t5_large_multi": "t5-large",
14+
"t5_small_multi": "google-t5/t5-small",
15+
"t5_base_multi": "google-t5/t5-base",
16+
"t5_large_multi": "google-t5/t5-large",
1717
"t5_1.1_small": "google/t5-v1_1-small",
1818
"t5_1.1_base": "google/t5-v1_1-base",
1919
"t5_1.1_large": "google/t5-v1_1-large",

0 commit comments

Comments
 (0)