Skip to content

Commit

Permalink
hard code model lists
Browse files Browse the repository at this point in the history
Signed-off-by: Chen Cui <[email protected]>
  • Loading branch information
cuichenx committed Jul 9, 2024
1 parent cfa1041 commit dbea3c1
Showing 1 changed file with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -210,9 +210,9 @@ def get_huggingface_pretrained_lm_models_list(
Returns the list of HuggingFace models
"""
logging.warning(
"pretrained_model_list no longer exists in HF transformers v4.40 and later. "
"See https://github.com/huggingface/transformers/pull/29112 for details. "
"`get_huggingface_pretrained_lm_models_list()` will return an empty list."
)
return []

huggingface_models = []
for model in HUGGINGFACE_MODELS:
model_names = HUGGINGFACE_MODELS[model]["pretrained_model_list"]
huggingface_models.extend(model_names)
return huggingface_models

0 comments on commit dbea3c1

Please sign in to comment.