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

[`8bit`] Fix 8bit corner case with Blip2 8bit (#25047)

fix 8bit corner case with Blip2 8bit
parent 3611fc90
...@@ -273,7 +273,7 @@ def get_keys_to_not_convert(model): ...@@ -273,7 +273,7 @@ def get_keys_to_not_convert(model):
return [] return []
# otherwise they have an attached head # otherwise they have an attached head
list_modules = list(model.named_children()) list_modules = list(model.named_parameters())
list_last_module = [list_modules[-1][0]] list_last_module = [list_modules[-1][0]]
# add last module together with tied weights # add last module together with tied 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