mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-04 16:22:57 +08:00
cp PR#2820 to release/2.0.2 (#2839)
This commit is contained in:
@@ -56,7 +56,7 @@ class ModelForCasualLM(nn.Layer, ABC):
|
||||
ori_vocab_size, use_topp_sampling, etc.
|
||||
"""
|
||||
super(ModelForCasualLM, self).__init__()
|
||||
|
||||
self.fd_config = configs
|
||||
@abstractmethod
|
||||
def set_state_dict(self, state_dict: Dict[str, Union[np.ndarray,
|
||||
paddle.Tensor]]):
|
||||
|
Reference in New Issue
Block a user