Skip to content

Commit d0f1bfa

Browse files
committed
feat: added comments to toml
1 parent 57474e5 commit d0f1bfa

File tree

4 files changed

+279
-138
lines changed

4 files changed

+279
-138
lines changed

cmd/project/defaults.go

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
package project
2+
3+
// Returns default model name for the example model type
4+
func getDefaultModelName(modelType string) string {
5+
switch modelType {
6+
case "LLM":
7+
return "google/flan-t5-base"
8+
case "Stable Diffusion":
9+
return "stabilityai/sdxl-turbo"
10+
case "Text to Audio":
11+
return "facebook/musicgen-small"
12+
}
13+
14+
return ""
15+
}

cmd/project/example.toml

-32
This file was deleted.

0 commit comments

Comments
 (0)