diff --git a/py/services/checkpoint_service.py b/py/services/checkpoint_service.py index d992db97..d110abd5 100644 --- a/py/services/checkpoint_service.py +++ b/py/services/checkpoint_service.py @@ -69,6 +69,8 @@ class CheckpointService(BaseModelService): "version_count": model_data.get("version_count"), "source_platform": model_data.get("source_platform", ""), "source_url": model_data.get("source_url", ""), + "source_model_id": model_data.get("source_model_id", ""), + "source_version_id": model_data.get("source_version_id", ""), "hf_url": model_data.get("hf_url", ""), } diff --git a/py/services/embedding_service.py b/py/services/embedding_service.py index 6d6baec1..d5fe28f2 100644 --- a/py/services/embedding_service.py +++ b/py/services/embedding_service.py @@ -69,6 +69,8 @@ class EmbeddingService(BaseModelService): "version_count": model_data.get("version_count"), "source_platform": model_data.get("source_platform", ""), "source_url": model_data.get("source_url", ""), + "source_model_id": model_data.get("source_model_id", ""), + "source_version_id": model_data.get("source_version_id", ""), "hf_url": model_data.get("hf_url", ""), } diff --git a/py/services/lora_service.py b/py/services/lora_service.py index d5e51858..91ea8579 100644 --- a/py/services/lora_service.py +++ b/py/services/lora_service.py @@ -81,6 +81,8 @@ class LoraService(BaseModelService): "version_count": model_data.get("version_count"), "source_platform": model_data.get("source_platform", ""), "source_url": model_data.get("source_url", ""), + "source_model_id": model_data.get("source_model_id", ""), + "source_version_id": model_data.get("source_version_id", ""), "hf_url": model_data.get("hf_url", ""), } diff --git a/py/services/other_model_service.py b/py/services/other_model_service.py index f5e47ccd..8a43d159 100644 --- a/py/services/other_model_service.py +++ b/py/services/other_model_service.py @@ -69,6 +69,8 @@ class OtherModelService(BaseModelService): "version_count": model_data.get("version_count"), "source_platform": model_data.get("source_platform", ""), "source_url": model_data.get("source_url", ""), + "source_model_id": model_data.get("source_model_id", ""), + "source_version_id": model_data.get("source_version_id", ""), "hf_url": model_data.get("hf_url", ""), }