mirror of
https://github.com/vladmandic/automatic
synced 2026-09-07 21:40:46 +02:00
Disty0
8 lines
270 B
Python
8 lines
270 B
Python
# pylint: disable=relative-beyond-top-level,redefined-builtin,protected-access
|
|
|
|
import torch
|
|
|
|
|
|
def quantized_linear_forward(self, input: torch.FloatTensor) -> torch.FloatTensor:
|
|
return torch.nn.functional.linear(input, self.sdnq_dequantizer(self.weight), self.bias)
|