a  d@s@dZddlZddlmmZddlmZGdddejZdS)z% Linear layer (alternate definition) N)nnc@s$eZdZdZejejdddZdS)LinearzApplies a linear transformation to the incoming data: :math:`y = xA^T + b` Wraps torch.nn.Linear to support AMP + torchscript usage by manually casting weight & bias to input.dtype to work around an issue w/ torch.addmm in this use case. )inputreturncCsZtjrD|jdur$|jj|jdnd}tj||jj|jd|dSt||j|jSdS)N)dtype)bias) torchZjitZ is_scriptingrtorFZlinearZweight)selfrrr [/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/layers/linear.pyforwards zLinear.forwardN)__name__ __module__ __qualname____doc__rZTensorrr r r r rsr)rrZtorch.nn.functionalrZ functionalr rr r r r s