theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

improve train_log.txt creation

Browse Source
This commit is contained in:
josc146 2023-12-22 13:00:13 +08:00
parent b988694da7
commit 2e0a542f33
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend-golang/app.go
View File

@ -50,9 +50,12 @@ func (a *App) OnStartup(ctx context.Context) {
os.Mkdir(a.exDir+"models", os.ModePerm) os.Mkdir(a.exDir+"models", os.ModePerm)
os.Mkdir(a.exDir+"lora-models", os.ModePerm) os.Mkdir(a.exDir+"lora-models", os.ModePerm)
os.Mkdir(a.exDir+"finetune/json2binidx_tool/data", os.ModePerm) os.Mkdir(a.exDir+"finetune/json2binidx_tool/data", os.ModePerm)
f, err := os.Create(a.exDir + "lora-models/train_log.txt") trainLogPath := a.exDir + "lora-models/train_log.txt"
if err == nil { if !a.FileExists(trainLogPath) {
f.Close() f, err := os.Create(trainLogPath)
if err == nil {
f.Close()
}
} }
a.downloadLoop() a.downloadLoop()