Skip to content

GitLab

Unverified Commit 81d82e4f authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

fix nasty `bnb` bug (#20408)

parent 658e5d8f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_utils.py
View file @ 81d82e4f
...@@ -2351,7 +2351,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -2351,7 +2351,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
load_in_8bit=load_in_8bit, load_in_8bit=load_in_8bit,
) )
cls.is_loaded_in_8bit = load_in_8bit model.is_loaded_in_8bit = load_in_8bit
# make sure token embedding weights are still tied if needed # make sure token embedding weights are still tied if needed
model.tie_weights() model.tie_weights()
......
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