Added support for caching text embeddings. This is just initial support and will probably fail for some models. Still needs to be ompimized

This commit is contained in:
Jaret Burkett
2025-08-07 10:27:55 -06:00
parent 4c4a10d439
commit bb6db3d635
16 changed files with 485 additions and 195 deletions

View File

@@ -110,6 +110,7 @@ export interface TrainConfig {
ema_config?: EMAConfig;
dtype: string;
unload_text_encoder: boolean;
cache_text_embeddings: boolean;
optimizer_params: {
weight_decay: number;
};