78cf049c29
* Base ace step 1.5 xl added. Generating, still wip on training and ui * Base training code done * Fix some issues with caching text embeddings. Update sample cards to show audio * Fix issue with quantizing ace step * Add album artwork to samples with waveform. * Cleanup logs * Add album art endpoint to speed up album art loading * Made an make video with artwork script * Make ui handle basic audio models. Make multi line adjustments to the editor and better syntax hilighting. * Add prompt tagging system for special tagged models. * prompt tagging processing for ui working. * Moved default samples to a special file so we can add more when needed and they can be adjusted for a specific model * Add a captioner job with music captioner that is prepped for use with the ui * Add basit ui setup for captioning modal and handeling captioning jobs * Starting captioning job from ui working. Still better management for it. * Better filtering of job options in the job view for captioning jobs * Added qwen3 vl as a captioner for images * Have an indicator when a dataset is being captioned. * Adjust the way caption jobs look in the queue * Fix a few issues. Adjust defaults. * Version bump * Added ace step to the readme.
48 lines
1.2 KiB
Plaintext
48 lines
1.2 KiB
Plaintext
generator client {
|
|
provider = "prisma-client-js"
|
|
}
|
|
|
|
datasource db {
|
|
provider = "sqlite"
|
|
url = "file:../../aitk_db.db"
|
|
}
|
|
|
|
model Settings {
|
|
id Int @id @default(autoincrement())
|
|
key String @unique
|
|
value String
|
|
}
|
|
|
|
model Queue {
|
|
id Int @id @default(autoincrement())
|
|
gpu_ids String @unique
|
|
is_running Boolean @default(false)
|
|
|
|
@@index([gpu_ids])
|
|
}
|
|
|
|
model Job {
|
|
id String @id @default(uuid())
|
|
name String @unique
|
|
gpu_ids String
|
|
job_config String // JSON string
|
|
created_at DateTime @default(now())
|
|
updated_at DateTime @updatedAt
|
|
status String @default("stopped")
|
|
stop Boolean @default(false)
|
|
return_to_queue Boolean @default(false) // same as stop, but will be set to 'queued' when stopped
|
|
step Int @default(0)
|
|
info String @default("")
|
|
speed_string String @default("")
|
|
queue_position Int @default(0)
|
|
pid Int?
|
|
job_type String @default("train") // 'train', 'caption'
|
|
job_ref String? // can be used for anything for special jobs, like dataset path for caption jobs
|
|
|
|
@@index([status])
|
|
@@index([gpu_ids])
|
|
@@index([job_type])
|
|
@@index([job_ref])
|
|
|
|
}
|