feat: Add Anima support (#860)
* Add Anima training support * Update Anima modular training * Use sample guidance for Anima * Fix Anima sampling * Limit Anima LoRA targets * Convert Anima LoRA exports * Fix Anima local loading * Update Anima default model * Pin upstream Anima diffusers * Adjust template defaults to be consistent with other models. Update README --------- Co-authored-by: Jaret Burkett (Ostris) <jaretburkett@gmail.com>
This commit is contained in:
@@ -63,9 +63,31 @@ export interface ModelArch {
|
||||
}
|
||||
|
||||
const defaultNameOrPath = '';
|
||||
const defaultLinearRank = 32
|
||||
const defaultLinearRank = 32;
|
||||
|
||||
export const modelArchs: ModelArch[] = [
|
||||
{
|
||||
name: 'anima',
|
||||
label: 'Anima',
|
||||
group: 'image',
|
||||
defaults: {
|
||||
// default updates when [selected, unselected] in the UI
|
||||
'config.process[0].model.name_or_path': ['circlestone-labs/Anima-Base-v1.0-Diffusers', defaultNameOrPath],
|
||||
'config.process[0].model.quantize': [false, false],
|
||||
'config.process[0].model.quantize_te': [false, false],
|
||||
'config.process[0].model.qtype': ['', 'qfloat8'],
|
||||
'config.process[0].model.qtype_te': ['', 'qfloat8'],
|
||||
'config.process[0].sample.sampler': ['flowmatch', 'flowmatch'],
|
||||
'config.process[0].train.noise_scheduler': ['flowmatch', 'flowmatch'],
|
||||
'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
|
||||
'config.process[0].sample.neg': [
|
||||
'worst quality, low quality, score_1, score_2, score_3, blurry, jpeg artifacts, sepia, signature, artist name',
|
||||
'',
|
||||
],
|
||||
},
|
||||
disableSections: ['network.conv'],
|
||||
additionalSections: ['model.low_vram', 'model.layer_offloading'],
|
||||
},
|
||||
{
|
||||
name: 'flux',
|
||||
label: 'FLUX.1',
|
||||
|
||||
Reference in New Issue
Block a user