Skip to content

Commit

Permalink
Fixed config.json download to go to user-supplied cache directory (#3…
Browse files Browse the repository at this point in the history
…0189)

* Fixed config.json download to go to user-supplied cache directory.

* Simplied implementation suggested by @amyeroberts
  • Loading branch information
ulatekh authored Apr 12, 2024
1 parent db7d155 commit b109257
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/transformers/pipelines/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -782,6 +782,7 @@ def pipeline(
_raise_exceptions_for_gated_repo=False,
_raise_exceptions_for_missing_entries=False,
_raise_exceptions_for_connection_errors=False,
cache_dir=model_kwargs.get("cache_dir"),
**hub_kwargs,
)
hub_kwargs["_commit_hash"] = extract_commit_hash(resolved_config_file, commit_hash)
Expand Down

0 comments on commit b109257

Please sign in to comment.