deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

refactoring

Browse Source
This commit is contained in:
dbaranchuk 2022-09-11 06:28:17 +03:00
parent d358999e9e
commit 4dd475ced4
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
bitsandbytes/autograd/_functions.py
View File

@ -1,6 +1,5 @@
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

2
bitsandbytes/nn/modules.py
View File

@ -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