convert : fix flake8 lint regarding lamdba assigment
This commit is contained in:
parent
78094fc4a7
commit
4c0170e206
|
|
@ -5416,8 +5416,7 @@ class LazyTorchTensor(gguf.LazyBase):
|
|||
dtype = cls._dtype_str_map[remote_tensor.dtype]
|
||||
shape = remote_tensor.shape
|
||||
meta = cls.meta_with_dtype_and_shape(dtype, shape)
|
||||
func = lambda r: torch.frombuffer(r.data(), dtype=dtype).reshape(shape)
|
||||
lazy = cls(meta=meta, args=(remote_tensor,), func=func)
|
||||
lazy = cls(meta=meta, args=(remote_tensor,), func=lambda r: torch.frombuffer(r.data(), dtype=dtype).reshape(shape))
|
||||
return cast(torch.Tensor, lazy)
|
||||
|
||||
@classmethod
|
||||
|
|
|
|||
Loading…
Reference in New Issue