feat: added comments to toml

runpod · Feb 12, 2024 · d0f1bfa · d0f1bfa
1 parent 57474e5
commit d0f1bfa
Show file tree

Hide file tree

Showing 4 changed files with 279 additions and 138 deletions.
diff --git a/cmd/project/defaults.go b/cmd/project/defaults.go
@@ -0,0 +1,15 @@
+package project
+
+// Returns default model name for the example model type
+func getDefaultModelName(modelType string) string {
+	switch modelType {
+	case "LLM":
+		return "google/flan-t5-base"
+	case "Stable Diffusion":
+		return "stabilityai/sdxl-turbo"
+	case "Text to Audio":
+		return "facebook/musicgen-small"
+	}
+
+	return ""
+}
diff --git a/cmd/project/example.toml b/cmd/project/example.toml