Skip to content

Commit eb30833

Browse files
committed
fix type annotation
1 parent 0f69807 commit eb30833

File tree

1 file changed

+1
-1
lines changed
  • src/diffusers/quantizers/bitsandbytes

1 file changed

+1
-1
lines changed

src/diffusers/quantizers/bitsandbytes/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def replace_with_bnb_linear(model, modules_to_not_convert=None, current_key_name
154154

155155

156156
# Copied from PEFT: https://github.com/huggingface/peft/blob/47b3712898539569c02ec5b3ed4a6c36811331a1/src/peft/utils/integrations.py#L41
157-
def dequantize_bnb_weight(weight: "torch.nn.Parameter", state=None, dtype: torch.dtype = None):
157+
def dequantize_bnb_weight(weight: "torch.nn.Parameter", state=None, dtype: "torch.dtype" = None):
158158
"""
159159
Helper function to dequantize 4bit or 8bit bnb weights.
160160

0 commit comments

Comments
 (0)