"git@developer.sourcefind.cn:OpenDAS/autoawq_kernels.git" did not exist on "54638dc2acb8af5aeb186c1c9b1e7cec9b480e23"
Unverified Commit f1ef74f8 authored by Tim Dettmers's avatar Tim Dettmers Committed by GitHub
Browse files

Merge pull request #859 from TimDettmers/younesbelkada-patch-1

Fix typehint error
parents fab00e26 67adb47b
...@@ -13,7 +13,7 @@ from scipy.stats import norm ...@@ -13,7 +13,7 @@ from scipy.stats import norm
import numpy as np import numpy as np
from functools import reduce # Required in Python 3 from functools import reduce # Required in Python 3
from typing import Tuple, Any from typing import Tuple, Any, Dict
from torch import Tensor from torch import Tensor
from bitsandbytes.utils import pack_dict_to_tensor, unpack_tensor_to_dict from bitsandbytes.utils import pack_dict_to_tensor, unpack_tensor_to_dict
...@@ -600,7 +600,7 @@ class QuantState: ...@@ -600,7 +600,7 @@ class QuantState:
return list_repr[idx] return list_repr[idx]
@classmethod @classmethod
def from_dict(cls, qs_dict: dict[str, Any], device: torch.device) -> 'QuantState': def from_dict(cls, qs_dict: Dict[str, Any], device: torch.device) -> 'QuantState':
""" """
unpacks components of state_dict into QuantState unpacks components of state_dict into QuantState
where necessary, convert into strings, torch.dtype, ints, etc. where necessary, convert into strings, torch.dtype, ints, etc.
......
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