Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
531b15c
1
Parent(s):
768fc4b
Add support to Qwen Image Edit Plus
Browse files
ui/src/app/jobs/new/options.ts
CHANGED
|
@@ -2,11 +2,21 @@ import { GroupedSelectOption, SelectOption } from '@/types';
|
|
| 2 |
|
| 3 |
type Control = 'depth' | 'line' | 'pose' | 'inpaint';
|
| 4 |
|
| 5 |
-
type DisableableSections =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
type AdditionalSections =
|
| 7 |
| 'datasets.control_path'
|
|
|
|
| 8 |
| 'datasets.do_i2v'
|
| 9 |
| 'sample.ctrl_img'
|
|
|
|
| 10 |
| 'datasets.num_frames'
|
| 11 |
| 'model.multistage'
|
| 12 |
| 'model.low_vram';
|
|
@@ -327,6 +337,28 @@ export const modelArchs: ModelArch[] = [
|
|
| 327 |
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_edit_torchao_uint3.safetensors',
|
| 328 |
},
|
| 329 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 330 |
{
|
| 331 |
name: 'hidream',
|
| 332 |
label: 'HiDream',
|
|
@@ -344,6 +376,9 @@ export const modelArchs: ModelArch[] = [
|
|
| 344 |
},
|
| 345 |
disableSections: ['network.conv'],
|
| 346 |
additionalSections: ['model.low_vram'],
|
|
|
|
|
|
|
|
|
|
| 347 |
},
|
| 348 |
{
|
| 349 |
name: 'hidream_e1',
|
|
|
|
| 2 |
|
| 3 |
type Control = 'depth' | 'line' | 'pose' | 'inpaint';
|
| 4 |
|
| 5 |
+
type DisableableSections =
|
| 6 |
+
| 'model.quantize'
|
| 7 |
+
| 'train.timestep_type'
|
| 8 |
+
| 'network.conv'
|
| 9 |
+
| 'trigger_word'
|
| 10 |
+
| 'train.diff_output_preservation'
|
| 11 |
+
| 'train.unload_text_encoder'
|
| 12 |
+
| 'slider';
|
| 13 |
+
|
| 14 |
type AdditionalSections =
|
| 15 |
| 'datasets.control_path'
|
| 16 |
+
| 'datasets.multi_control_paths'
|
| 17 |
| 'datasets.do_i2v'
|
| 18 |
| 'sample.ctrl_img'
|
| 19 |
+
| 'sample.multi_ctrl_imgs'
|
| 20 |
| 'datasets.num_frames'
|
| 21 |
| 'model.multistage'
|
| 22 |
| 'model.low_vram';
|
|
|
|
| 337 |
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_edit_torchao_uint3.safetensors',
|
| 338 |
},
|
| 339 |
},
|
| 340 |
+
{
|
| 341 |
+
name: 'qwen_image_edit_plus',
|
| 342 |
+
label: 'Qwen-Image-Edit-2509',
|
| 343 |
+
group: 'instruction',
|
| 344 |
+
defaults: {
|
| 345 |
+
// default updates when [selected, unselected] in the UI
|
| 346 |
+
'config.process[0].model.name_or_path': ['Qwen/Qwen-Image-Edit-2509', defaultNameOrPath],
|
| 347 |
+
'config.process[0].model.quantize': [true, false],
|
| 348 |
+
'config.process[0].model.quantize_te': [true, false],
|
| 349 |
+
'config.process[0].model.low_vram': [true, false],
|
| 350 |
+
'config.process[0].train.unload_text_encoder': [false, false],
|
| 351 |
+
'config.process[0].sample.sampler': ['flowmatch', 'flowmatch'],
|
| 352 |
+
'config.process[0].train.noise_scheduler': ['flowmatch', 'flowmatch'],
|
| 353 |
+
'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
|
| 354 |
+
'config.process[0].model.qtype': ['qfloat8', 'qfloat8'],
|
| 355 |
+
},
|
| 356 |
+
disableSections: ['network.conv', 'train.unload_text_encoder'],
|
| 357 |
+
additionalSections: ['datasets.multi_control_paths', 'sample.multi_ctrl_imgs', 'model.low_vram'],
|
| 358 |
+
accuracyRecoveryAdapters: {
|
| 359 |
+
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_edit_2509_torchao_uint3.safetensors',
|
| 360 |
+
},
|
| 361 |
+
},
|
| 362 |
{
|
| 363 |
name: 'hidream',
|
| 364 |
label: 'HiDream',
|
|
|
|
| 376 |
},
|
| 377 |
disableSections: ['network.conv'],
|
| 378 |
additionalSections: ['model.low_vram'],
|
| 379 |
+
accuracyRecoveryAdapters: {
|
| 380 |
+
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/hidream_i1_full_torchao_uint3.safetensors',
|
| 381 |
+
},
|
| 382 |
},
|
| 383 |
{
|
| 384 |
name: 'hidream_e1',
|