From 75d7ebe1462427e75cc508a836188fc66638ae69 Mon Sep 17 00:00:00 2001
From: anda-ren <renyanda@starwhale.ai>
Date: Mon, 13 Nov 2023 14:16:47 +0800
Subject: [PATCH] model copy with ft id

---
 .../java/ai/starwhale/mlops/domain/model/ModelService.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/server/controller/src/main/java/ai/starwhale/mlops/domain/model/ModelService.java b/server/controller/src/main/java/ai/starwhale/mlops/domain/model/ModelService.java
index 9fed6f82da..3bb3ce7e4b 100644
--- a/server/controller/src/main/java/ai/starwhale/mlops/domain/model/ModelService.java
+++ b/server/controller/src/main/java/ai/starwhale/mlops/domain/model/ModelService.java
@@ -597,7 +597,12 @@ public void createModelVersion(
                     .storageSize(storageSize)
                     .build();
             this.modelVersionMapper.insert(modelVersionEntity);
-            fineTuneService.attachTargetModel(createModelVersionRequest.getModelSource().getId(), modelVersionEntity);
+            if (sourcedFromFineTune) {
+                fineTuneService.attachTargetModel(
+                        createModelVersionRequest.getModelSource().getId(),
+                        modelVersionEntity
+                );
+            }
             RevertManager.create(this.bundleManager, this.modelDao)
                     .revertVersionTo(modelVersionEntity.getModelId(), modelVersionEntity.getId());
         } else {