File tree Expand file tree Collapse file tree 2 files changed +2
-6
lines changed
src/compressed_tensors/transform/factory Expand file tree Collapse file tree 2 files changed +2
-6
lines changed Original file line number Diff line number Diff line change 14
14
15
15
from abc import ABC , abstractmethod
16
16
from collections import defaultdict
17
- from typing import List , Optional , Tuple , Set
17
+ from typing import List , Optional , Set , Tuple
18
18
19
19
import torch
20
20
import torch .nn .utils .parametrize as P
21
- from compressed_tensors .utils .internal import InternalModule
22
21
from compressed_tensors .registry .registry import RegistryMixin , T
23
22
from compressed_tensors .transform import (
24
23
TransformArgs ,
34
33
register_offload_module ,
35
34
update_offload_parameter ,
36
35
)
36
+ from compressed_tensors .utils .internal import InternalModule
37
37
from torch import Tensor
38
38
from torch .nn import Module , Parameter
39
39
Original file line number Diff line number Diff line change 13
13
# limitations under the License.
14
14
15
15
import math
16
- < << << << HEAD
17
16
from typing import Optional
18
- == == == =
19
- from typing import Optional , Union
20
- > >> >> >> kylesayrs / serialize - tconfig
21
17
22
18
import torch
23
19
from compressed_tensors .transform import TransformArgs , TransformScheme
You can’t perform that action at this time.
0 commit comments