feat: update modelfile commands to uppercase format (#118)

Signed-off-by: Gaius <[email protected]>

Author: gaius-qi

docs/getting-started.md

@@ -39,43 +39,40 @@ Example of Modelfile:
 
 ```shell
 # Model name (string), such as llama3-8b-instruct, gpt2-xl, qwen2-vl-72b-instruct, etc.
-name gemma-2b
+NAME gemma-2b
 
 # Model architecture (string), such as transformer, cnn, rnn, etc.
-arch transformer
+ARCH transformer
 
 # Model family (string), such as llama3, gpt2, qwen2, etc.
-family gemma
+FAMILY gemma
 
 # Model format (string), such as onnx, tensorflow, pytorch, etc.
-format safetensors
+FORMAT safetensors
 
 # Number of parameters in the model (integer).
-paramsize 16
+PARAMSIZE 16
 
 # Model precision (string), such as bf16, fp16, int8, etc.
-precision bf16
+PRECISION bf16
 
 # Model quantization (string), such as awq, gptq, etc.
-quantization awq
+QUANTIZATION awq
 
 # Specify model configuration file, support glob path pattern.
-config config.json
+CONFIG config.json
 
 # Specify model configuration file, support glob path pattern.
-config generation_config.json
+CONFIG generation_config.json
 
 # Model weight, support glob path pattern.
-model *.safetensors
+MODEL *.safetensors
 
 # Specify code, support glob path pattern.
-code *.py
+CODE *.py
 
 # Specify documentation, support glob path pattern.
-doc *.md
-
-# Specify dataset, support glob path pattern.
-dataset *.csv
+DOC *.md
 ```
 
 Then run the following command to build the model artifact:

pkg/modelfile/command/command.go

@@ -22,58 +22,58 @@ const (
 	// the model to be served, such as the config.json, generation_config.json, etc.
 	// The CONFIG command can be used multiple times in a modelfile, it
 	// will be copied the config file to the artifact package as a layer.
-	CONFIG = "config"
+	CONFIG = "CONFIG"
 
 	// MODEL is the command to set the model file path. The value of this command
 	// is the glob of the model file path to match the model file name.
 	// The MODEL command can be used multiple times in a modelfile, it will scan
 	// the model file path by the glob and copy each model file to the artifact
 	// package, and each model file will be a layer.
-	MODEL = "model"
+	MODEL = "MODEL"
 
 	// CODE is the command to set the code file path. The value of this commands
 	// is the glob of the code file path to match the code file name.
 	// The CODE command can be used multiple times in a modelfile, it will scan
 	// the code file path by the glob and copy each code file to the artifact
 	// package, and each code file will be a layer.
-	CODE = "code"
+	CODE = "CODE"
 
 	// DATASET is the command to set the dataset file path. The value of this commands
 	// is the glob of the dataset file path to match the dataset file name.
 	// The DATASET command can be used multiple times in a modelfile, it will scan
 	// the dataset file path by the glob and copy each dataset file to the artifact
 	// package, and each dataset file will be a layer.
-	DATASET = "dataset"
+	DATASET = "DATASET"
 
 	// DOC is the command to set the documentation file path. The value of this commands
 	// is the glob of the documentation file path to match the documentation file name.
 	// The DOC command can be used multiple times in a modelfile, it will scan
 	// the documentation file path by the glob and copy each documentation file to the artifact
 	// package, and each documentation file will be a layer.
-	DOC = "doc"
+	DOC = "DOC"
 
 	// NAME is the command to set the model name, such as llama3-8b-instruct, gpt2-xl,
 	// qwen2-vl-72b-instruct, etc.
-	NAME = "name"
+	NAME = "NAME"
 
 	// ARCH is the command to set the architecture of the model, such as transformer,
 	// cnn, rnn, etc.
-	ARCH = "arch"
+	ARCH = "ARCH"
 
 	// FAMILY is the command to set the family of the model, such as llama3, gpt2, qwen2, etc.
-	FAMILY = "family"
+	FAMILY = "FAMILY"
 
 	// FORMAT is the command to set the format of the model, such as onnx, tensorflow, pytorch, etc.
-	FORMAT = "format"
+	FORMAT = "FORMAT"
 
 	// PARAMSIZE is the command to set the parameter size of the model.
-	PARAMSIZE = "paramsize"
+	PARAMSIZE = "PARAMSIZE"
 
 	// PRECISION is the command to set the precision of the model, such as bf16, fp16, int8, etc.
-	PRECISION = "precision"
+	PRECISION = "PRECISION"
 
 	// QUANTIZATION is the command to set the quantization of the model, such as awq, gptq, etc.
-	QUANTIZATION = "quantization"
+	QUANTIZATION = "QUANTIZATION"
 )
 
 // Commands is a list of all the commands that can be used in a modelfile.

pkg/modelfile/constants.go

@@ -116,6 +116,7 @@ func isFileType(filename string, patterns []string) bool {
 			return true
 		}
 	}
+
 	return false
 }
 
@@ -135,5 +136,6 @@ func isSkippable(filename string) bool {
 			return true
 		}
 	}
+
 	return false
 }