Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "abbf3c1adf6bbf424ae82e640647ad4078e1a4b9"
Commit 4dd475ce authored by dbaranchuk's avatar dbaranchuk
Browse files

refactoring

parent d358999e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
bitsandbytes/autograd/_functions.py
View file @ 4dd475ce
import operator import operator
import torch import torch
import bitsandbytes as bnb
import bitsandbytes.functional as F import bitsandbytes.functional as F
from dataclasses import dataclass from dataclasses import dataclass
......
bitsandbytes/nn/modules.py
View file @ 4dd475ce
...@@ -221,9 +221,9 @@ class Linear8bitLt(nn.Linear): ...@@ -221,9 +221,9 @@ class Linear8bitLt(nn.Linear):
output_features, output_features,
bias=True, bias=True,
has_fp16_weights=True, has_fp16_weights=True,
memory_efficient_backward=False,
threshold=0.0, threshold=0.0,
index=None, index=None,
memory_efficient_backward=False
): ):
super(Linear8bitLt, self).__init__( super(Linear8bitLt, self).__init__(
input_features, output_features, bias input_features, output_features, bias
......
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