We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b3b3bcc commit d1049e7Copy full SHA for d1049e7
1 file changed
bitsandbytes/nn/modules.py
@@ -434,7 +434,7 @@ class Linear4bit(nn.Linear):
434
import torch.nn as nn
435
436
import bitsandbytes as bnb
437
- from bnb.nn import Linear4bit
+ from bitsandbytes.nn import Linear4bit
438
439
fp16_model = nn.Sequential(
440
nn.Linear(64, 64),
@@ -948,7 +948,7 @@ class Linear8bitLt(nn.Linear):
948
949
950
951
- from bnb.nn import Linear8bitLt
+ from bitsandbytes.nn import Linear8bitLt
952
953
954
0 commit comments