ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Fix name device is not found bug (#12703)

Browse source
This commit is contained in:
Yuwen Hu 2025-01-13 10:11:02 +08:00 committed by GitHub
parent e2d58f733e
commit a1da7908b9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/src/ipex_llm/transformers/low_bit_linear.py
View file

@ -395,7 +395,7 @@ class FP4Params(torch.nn.Parameter):
return self._shape
@overload
def to(self: T, device: Optional[Union[int, device]]=...,
def to(self: T, device: Optional[Union[int, torch.device]]=...,
dtype: Optional[Union[dtype, str]]=..., non_blocking: bool=...,) -> T:
...