mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-30 11:11:37 +00:00
Added initial support for Qwen-Image-2512
This commit is contained in:
@@ -327,6 +327,28 @@ export const modelArchs: ModelArch[] = [
|
|||||||
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors',
|
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors',
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
name: 'qwen_image:2512',
|
||||||
|
label: 'Qwen-Image-2512',
|
||||||
|
group: 'image',
|
||||||
|
defaults: {
|
||||||
|
// default updates when [selected, unselected] in the UI
|
||||||
|
'config.process[0].model.name_or_path': ['Qwen/Qwen-Image-2512', defaultNameOrPath],
|
||||||
|
'config.process[0].model.quantize': [true, false],
|
||||||
|
'config.process[0].model.quantize_te': [true, false],
|
||||||
|
'config.process[0].model.low_vram': [true, false],
|
||||||
|
'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].model.qtype': ['qfloat8', 'qfloat8'],
|
||||||
|
},
|
||||||
|
disableSections: ['network.conv'],
|
||||||
|
additionalSections: ['model.low_vram', 'model.layer_offloading'],
|
||||||
|
// Training an ARA now, the other one will not work
|
||||||
|
// accuracyRecoveryAdapters: {
|
||||||
|
// '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors',
|
||||||
|
// },
|
||||||
|
},
|
||||||
{
|
{
|
||||||
name: 'qwen_image_edit',
|
name: 'qwen_image_edit',
|
||||||
label: 'Qwen-Image-Edit',
|
label: 'Qwen-Image-Edit',
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
VERSION = "0.7.14"
|
VERSION = "0.7.15"
|
||||||
|
|||||||
Reference in New Issue
Block a user