Update tiny model creation script and some others files (#22006)
* Update 1 * Update 2 * Update 3 * Update 4 * Update 5 * Update 6 * Update 7 * Update 8 * Update 9 * Update 10 --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -51,10 +51,12 @@ def get_checkpoint_from_config_class(config_class):
|
||||
config_source = inspect.getsource(config_class)
|
||||
checkpoints = _re_checkpoint.findall(config_source)
|
||||
|
||||
for checkpoint in checkpoints:
|
||||
# Each `checkpoint` is a tuple of a checkpoint name and a checkpoint link.
|
||||
# For example, `('bert-base-uncased', 'https://huggingface.co/bert-base-uncased')`
|
||||
ckpt_name, ckpt_link = checkpoint
|
||||
# Each `checkpoint` is a tuple of a checkpoint name and a checkpoint link.
|
||||
# For example, `('bert-base-uncased', 'https://huggingface.co/bert-base-uncased')`
|
||||
for ckpt_name, ckpt_link in checkpoints:
|
||||
# allow the link to end with `/`
|
||||
if ckpt_link.endswith("/"):
|
||||
ckpt_link = ckpt_link[:-1]
|
||||
|
||||
# verify the checkpoint name corresponds to the checkpoint link
|
||||
ckpt_link_from_name = f"https://huggingface.co/{ckpt_name}"
|
||||
|
||||
@@ -782,6 +782,11 @@ def get_config_overrides(config_class, processors):
|
||||
# CLIP-like models have `text_model_tester` and `vision_model_tester`, and we need to pass `vocab_size` to
|
||||
# `text_model_tester` via `text_kwargs`. The same trick is also necessary for `Flava`.
|
||||
if config_class.__name__ in [
|
||||
"AlignConfig",
|
||||
"AltCLIPConfig",
|
||||
"ChineseCLIPConfig",
|
||||
"CLIPSegConfig",
|
||||
"ClapConfig",
|
||||
"CLIPConfig",
|
||||
"GroupViTConfig",
|
||||
"OwlViTConfig",
|
||||
|
||||
Reference in New Issue
Block a user