8cf7c4d8ad
Signed-off-by: chengvjiang <chengvjiang@tencent.com> Co-authored-by: chengvjiang <chengvjiang@tencent.com> Co-authored-by: Andreas Karatzas <akaratza@amd.com>
19 lines
407 B
Python
19 lines
407 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
import torch.nn as nn
|
|
|
|
|
|
class HpcModule(nn.Module):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
@classmethod
|
|
def support(cls, *args, **kwargs):
|
|
return True
|
|
|
|
def process_weights_after_loading(self, model):
|
|
pass
|
|
|
|
def forward(self, *args, **kwargs):
|
|
pass
|