Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "dedc7a8fdbdf4b9824fa2943a43be4ca147733de"
Unverified Commit 63fb253d authored by James Braza's avatar James Braza Committed by GitHub
Browse files

Fixing `name 'torch' is not defined` in `bitsandbytes` integration (#31243) 

Fixed torch definition error
parent 66875ac0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/integrations/bitsandbytes.py
View file @ 63fb253d
...@@ -332,7 +332,7 @@ def get_keys_to_not_convert(model): ...@@ -332,7 +332,7 @@ def get_keys_to_not_convert(model):
# Copied from PEFT: https://github.com/huggingface/peft/blob/47b3712898539569c02ec5b3ed4a6c36811331a1/src/peft/utils/integrations.py#L41 # Copied from PEFT: https://github.com/huggingface/peft/blob/47b3712898539569c02ec5b3ed4a6c36811331a1/src/peft/utils/integrations.py#L41
def dequantize_bnb_weight(weight: torch.nn.Parameter, state=None): def dequantize_bnb_weight(weight: "torch.nn.Parameter", state=None):
""" """
Helper function to dequantize 4bit or 8bit bnb weights. Helper function to dequantize 4bit or 8bit bnb 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