"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ff066119ca361dfe19601d91067ec03561470435"
Unverified Commit 94d416f0 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

FIX: Add `accelerate` as a hard requirement (#31090)

add accelerate
parent 22dab246
...@@ -3048,6 +3048,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -3048,6 +3048,9 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
" ignored." " ignored."
) )
if gguf_file is not None and not is_accelerate_available():
raise ValueError("accelerate is required when loading a GGUF file `pip install accelerate`.")
if commit_hash is None: if commit_hash is None:
if not isinstance(config, PretrainedConfig): if not isinstance(config, PretrainedConfig):
# We make a call to the config file first (which may be absent) to get the commit hash as soon as possible # We make a call to the config file first (which may be absent) to get the commit hash as soon as possible
......
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