mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-15 11:02:52 +00:00
fixes
This commit is contained in:
parent
54354257dc
commit
bbd40d8cc4
3 changed files with 3 additions and 6 deletions
|
|
@ -102,9 +102,6 @@ class DiskDistributionRegistry(DistributionRegistry):
|
|||
"Unregister it first if you want to replace it."
|
||||
)
|
||||
|
||||
if "sentence-transformers/sentence-transformers" in obj.identifier:
|
||||
raise Exception("OMG")
|
||||
|
||||
await self.kvstore.set(
|
||||
KEY_FORMAT.format(type=obj.type, identifier=obj.identifier),
|
||||
obj.model_dump_json(),
|
||||
|
|
|
|||
|
|
@ -435,7 +435,7 @@ class OpenAIMixin(NeedsRequestProviderData, ABC, BaseModel):
|
|||
"""
|
||||
# First check if the model is pre-registered in the model store
|
||||
if hasattr(self, "model_store") and self.model_store:
|
||||
qualified_model = f"{self.__provider_id__}/{model}"
|
||||
qualified_model = f"{self.__provider_id__}/{model}" # type: ignore[attr-defined]
|
||||
if await self.model_store.has_model(qualified_model):
|
||||
return True
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue