add model wait

1a00717c · duanjinfei · 649a3e9a · 1a00717c
Commit 1a00717c authored May 24, 2024 by duanjinfei
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

model_handler.go largeModel/model_handler.go +3 -0

No files found.
--- a/largeModel/model_handler.go
+++ b/largeModel/model_handler.go
@@ -148,6 +148,9 @@ func (m *ModelHandler) MonitorModelStatus() {
 		select {
 		case <-ticker.C:
 			{
+				if !m.IsInit {
+					continue
+				}
 				log.Info("Start monitoring model status for update image info")
 				imageList, _ := m.dockerOp.PsImages()
 				if imageList != nil && len(imageList) > 0 {