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 ef70f2a commit 54bd6edCopy full SHA for 54bd6ed
bitsandbytes/nn/modules.py
@@ -2,24 +2,11 @@
2
#
3
# This source code is licensed under the MIT license found in the
4
# LICENSE file in the root directory of this source tree.
5
-from typing import (
6
- Any,
7
- Callable,
8
- Dict,
9
- Iterator,
10
- Mapping,
11
- Optional,
12
- Set,
13
- Tuple,
14
- TypeVar,
15
- Union,
16
- overload,
17
-)
+from typing import Optional, TypeVar, Union, overload
18
19
import torch
20
import torch.nn.functional as F
21
from torch import Tensor, device, dtype, nn
22
-from torch.nn.parameter import Parameter
23
24
import bitsandbytes as bnb
25
from bitsandbytes.optim import GlobalOptimManager
0 commit comments