[PT FE]: fix gptq patching for strided tensors (#24871)
### Details: - *fix reshaping of strided tensor in gptq patcher* ### Tickets: - *CVS-140119* - *CVS-143119*
This commit is contained in:
parent
9465604223
commit
73a2d13721
|
|
@ -32,7 +32,7 @@ def patched_forward(self, *args, **kwargs):
|
|||
x = args[0]
|
||||
dtype = x.dtype
|
||||
outshape = x.shape[:-1] + (self.width,)
|
||||
x = x.view(-1, x.shape[-1])
|
||||
x = x.contiguous().view(-1, x.shape[-1])
|
||||
groups = self.qzeros.shape[0]
|
||||
height = self.qweight.shape[0]
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue