Change auto_memory to be layer_offloading and allow you to set the amount to unload
This commit is contained in:
@@ -153,7 +153,9 @@ export interface ModelConfig {
|
||||
arch: string;
|
||||
low_vram: boolean;
|
||||
model_kwargs: { [key: string]: any };
|
||||
auto_memory?: boolean;
|
||||
layer_offloading?: boolean;
|
||||
layer_offloading_transformer_percent?: number;
|
||||
layer_offloading_text_encoder_percent?: number;
|
||||
}
|
||||
|
||||
export interface SampleItem {
|
||||
|
||||
Reference in New Issue
Block a user