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:
@@ -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;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user