fix(mm): fix typing on heuristic_import

This commit is contained in:
psychedelicious 2024-03-05 19:18:44 +11:00
parent 99407c899f
commit 656839fcd1

View File

@ -179,7 +179,7 @@ class ModelInstallService(ModelInstallServiceBase):
source: str, source: str,
config: Optional[Dict[str, Any]] = None, config: Optional[Dict[str, Any]] = None,
access_token: Optional[str] = None, access_token: Optional[str] = None,
inplace: bool = False, inplace: Optional[bool] = False,
) -> ModelInstallJob: ) -> ModelInstallJob:
variants = "|".join(ModelRepoVariant.__members__.values()) variants = "|".join(ModelRepoVariant.__members__.values())
hf_repoid_re = f"^([^/:]+/[^/:]+)(?::({variants})?(?::/?([^:]+))?)?$" hf_repoid_re = f"^([^/:]+/[^/:]+)(?::({variants})?(?::/?([^:]+))?)?$"