apolinario commited on
Commit
531b15c
·
1 Parent(s): 768fc4b

Add support to Qwen Image Edit Plus

Browse files
Files changed (1) hide show
  1. ui/src/app/jobs/new/options.ts +36 -1
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 = 'model.quantize' | 'train.timestep_type' | 'network.conv';
 
 
 
 
 
 
 
 
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',