model copy with ft id

star-whale · Nov 13, 2023 · 75d7ebe · 75d7ebe
1 parent c9a08fc
commit 75d7ebe
Showing 1 changed file with 6 additions and 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
@@ -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 {