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.
32 lines
786 B
Python
32 lines
786 B
Python
from toolkit.extension import Extension
|
|
|
|
|
|
class AceStepCaptionerExtension(Extension):
|
|
uid = "AceStepCaptioner"
|
|
name = "Ace Step Captioner"
|
|
|
|
@classmethod
|
|
def get_process(cls):
|
|
# import your process class here so it is only loaded when needed and return it
|
|
from .AceStepCaptioner import AceStepCaptioner
|
|
|
|
return AceStepCaptioner
|
|
|
|
|
|
class Qwen3VLCaptionerExtension(Extension):
|
|
uid = "Qwen3VLCaptioner"
|
|
name = "Qwen 3VL Captioner"
|
|
|
|
@classmethod
|
|
def get_process(cls):
|
|
# import your process class here so it is only loaded when needed and return it
|
|
from .Qwen3VLCaptioner import Qwen3VLCaptioner
|
|
|
|
return Qwen3VLCaptioner
|
|
|
|
|
|
AI_TOOLKIT_EXTENSIONS = [
|
|
AceStepCaptionerExtension,
|
|
Qwen3VLCaptionerExtension,
|
|
]
|