lora finetune (need to be refactored)

This commit is contained in:
josc146
2023-07-03 17:41:47 +08:00
parent c54d10795f
commit 987854fe49
42 changed files with 4825 additions and 158 deletions

10
main.go
View File

@@ -26,12 +26,22 @@ var cyacInfo embed.FS
//go:embed backend-python
var py embed.FS
//go:embed finetune
var finetune embed.FS
func main() {
if buildInfo, ok := debug.ReadBuildInfo(); !ok || strings.Contains(buildInfo.String(), "-ldflags") {
backend.CopyEmbed(cyac)
backend.CopyEmbed(cyacInfo)
backend.CopyEmbed(py)
backend.CopyEmbed(finetune)
os.Mkdir("models", os.ModePerm)
os.Mkdir("lora-models", os.ModePerm)
}
f, err := os.Create("lora-models/train_log.txt")
if err == nil {
f.Close()
}
// Create an instance of the app structure