|
57 | 57 | VLLM_INFERENCE_RESTRICTED_PARAMS, |
58 | 58 | ) |
59 | 59 | from ads.aqua.data import AquaResourceIdentifier |
60 | | -from ads.aqua.model.constants import ModelTask |
61 | 60 | from ads.common.auth import AuthState, default_signer |
62 | 61 | from ads.common.extended_enum import ExtendedEnumMeta |
63 | 62 | from ads.common.object_storage_details import ObjectStorageDetails |
64 | 63 | from ads.common.oci_resource import SEARCH_TYPE, OCIResource |
65 | 64 | from ads.common.utils import copy_file, get_console_link, upload_to_os |
66 | 65 | from ads.config import AQUA_SERVICE_MODELS_BUCKET, CONDA_BUCKET_NS, TENANCY_OCID |
67 | 66 | from ads.model import DataScienceModel, ModelVersionSet |
68 | | -from tests.unitary.with_extras.model.score import model_name |
69 | 67 |
|
70 | 68 | logger = logging.getLogger("ads.aqua") |
71 | 69 |
|
@@ -1065,11 +1063,5 @@ def get_hf_model_info(repo_id: str) -> ModelInfo: |
1065 | 1063 | except HfHubHTTPError as err: |
1066 | 1064 | raise format_hf_custom_error_message(err) from err |
1067 | 1065 |
|
1068 | | -def list_hf_models(query:str) -> List[str]: |
1069 | | - try: |
1070 | | - models= HfApi().list_models(model_name=query,task=ModelTask.TEXT_GENERATION) |
1071 | | - return [model.id for model in models if model.disabled is None] |
1072 | | - except HfHubHTTPError as err: |
1073 | | - raise format_hf_custom_error_message(err) from err |
1074 | 1066 |
|
1075 | 1067 |
|
0 commit comments