Add get weights by parameter name for llama (#2266)
This commit is contained in:
@@ -365,6 +365,17 @@ class UpdateWeightFromDiskReqOutput:
|
||||
message: str
|
||||
|
||||
|
||||
@dataclass
|
||||
class GetWeightsByNameReqInput:
|
||||
name: str
|
||||
truncate_size: int = 100
|
||||
|
||||
|
||||
@dataclass
|
||||
class GetWeightsByNameReqOutput:
|
||||
parameter: list
|
||||
|
||||
|
||||
@dataclass
|
||||
class AbortReq:
|
||||
# The request id
|
||||
|
||||
Reference in New Issue
Block a user