Skip to content

GitLab

Unverified Commit bfbec177 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix add-new-model-like when model doesn't support all frameworks (#16966)

parent cf8a7c24
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/commands/add_new_model_like.py
View file @ bfbec177
...@@ -517,7 +517,7 @@ def filter_framework_files( ...@@ -517,7 +517,7 @@ def filter_framework_files(
else: else:
framework_to_file["pt"] = f framework_to_file["pt"] = f
return [framework_to_file[f] for f in frameworks] + others return [framework_to_file[f] for f in frameworks if f in framework_to_file] + others
def get_model_files(model_type: str, frameworks: Optional[List[str]] = None) -> Dict[str, Union[Path, List[Path]]]: def get_model_files(model_type: str, frameworks: Optional[List[str]] = None) -> Dict[str, Union[Path, List[Path]]]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment