update modelx config yaml

Signed-off-by: dongjiang <[email protected]>
kubeservice-stack · Dec 24, 2024 · f497361 · f497361
1 parent 45ed5c9
commit f497361
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/cmd/modelx/model/config.go b/cmd/modelx/model/config.go
@@ -17,13 +17,15 @@ limitations under the License.
 package model
 
 type ModelConfig struct {
-	Description string            `json:"description"`
-	FrameWork   string            `json:"framework"`
-	Task        string            `json:"task"`
-	Tags        []string          `json:"tags"`
-	Resources   map[string]any    `json:"resources"`
-	Mantainers  []string          `json:"maintainers"`
-	Annotations map[string]string `json:"annotations,omitempty"`
-	ModelFiles  []string          `json:"modelFiles"`
-	Config      any               `json:"config"`
+	Description  string            `json:"description"`
+	FrameWork    string            `json:"framework"`
+	Command      []string          `json:"command,omitempty"`
+	Args         []string          `json:"args,omitempty"`
+	Tags         []string          `json:"tags"`
+	Resources    map[string]any    `json:"resources"`
+	Mantainers   []string          `json:"maintainers"`
+	Annotations  map[string]string `json:"annotations,omitempty"`
+	ModelFiles   []string          `json:"modelFiles"`
+	Config       any               `json:"config"`
+	Dependencies []*ModelConfig    `json:"dependencies,omitempty"`
 }
diff --git a/cmd/modelx/model/init.go b/cmd/modelx/model/init.go
@@ -74,8 +74,8 @@ func InitModelx(ctx context.Context, path string, force bool) error {
 		return fmt.Errorf("create modelx directory:%s %w", path, err)
 	}
 	config := ModelConfig{
-		Description: "This is a modelx model",
-		FrameWork:   "<some framework.Pytorch|TensorFlow|Caffe|MindSpore|etc>",
+		Description: "This is a modelx model.",
+		FrameWork:   "<Pytorch/TensorFlow/Caffe/MindSpore/etc>",
 		Config: map[string]interface{}{
 			"inputs":  map[string]interface{}{},
 			"outputs": map[string]interface{}{},
@@ -96,7 +96,9 @@ func InitModelx(ctx context.Context, path string, force bool) error {
 		Mantainers: []string{
 			"maintainer",
 		},
-		ModelFiles: []string{},
+		ModelFiles:   []string{},
+		Command:      []string{},
+		Dependencies: nil,
 	}
 	var configcontent bytes.Buffer
 	encoder := yaml.NewEncoder(&configcontent)