mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-08 19:10:56 +00:00
Make Safety test work, other cleanup
This commit is contained in:
parent
ba1f294cc6
commit
fcd22b6baa
16 changed files with 229 additions and 123 deletions
|
|
@ -63,19 +63,6 @@ class _HfAdapter(Inference, ModelsProtocolPrivate):
|
|||
)
|
||||
]
|
||||
|
||||
async def get_model(self, identifier: str) -> Optional[ModelDef]:
|
||||
model = self.huggingface_repo_to_llama_model_id.get(self.model_id)
|
||||
if model != identifier:
|
||||
return None
|
||||
|
||||
return ModelDef(
|
||||
identifier=model,
|
||||
llama_model=model,
|
||||
metadata={
|
||||
"huggingface_repo": self.model_id,
|
||||
},
|
||||
)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
pass
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue