apolinario commited on
Commit
cdc26bb
·
1 Parent(s): b170468

fix qwen image

Browse files
ui/src/app/api/hf-jobs/route.ts CHANGED
@@ -611,7 +611,7 @@ def generate_model_card_readme(repo_id: str, config: dict, model_name: str, cura
611
  lower_model_name = (model_config.get("name_or_path", "") or "").lower()
612
  base_model_lower = (base_model or "").lower()
613
 
614
- instruction_arches = {'flux_kontext', 'hidream_e1', 'qwen_image_edit'}
615
  is_instruction = lower_arch in instruction_arches or 'kontext' in lower_model_name
616
  is_edit_model = (
617
  'edit' in lower_arch
 
611
  lower_model_name = (model_config.get("name_or_path", "") or "").lower()
612
  base_model_lower = (base_model or "").lower()
613
 
614
+ instruction_arches = {'flux_kontext', 'hidream_e1', 'qwen_image_edit', 'qwen_image_edit_plus'}
615
  is_instruction = lower_arch in instruction_arches or 'kontext' in lower_model_name
616
  is_edit_model = (
617
  'edit' in lower_arch
ui/src/app/jobs/new/options.ts CHANGED
@@ -322,7 +322,7 @@ export const modelArchs: ModelArch[] = [
322
  'config.process[0].train.lr': [0.0003, 0.0001],
323
  },
324
  disableSections: ['network.conv'],
325
- additionalSections: ['datasets.control_path', 'sample.ctrl_img', 'model.low_vram'],
326
  accuracyRecoveryAdapters: {
327
  '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors',
328
  },
 
322
  'config.process[0].train.lr': [0.0003, 0.0001],
323
  },
324
  disableSections: ['network.conv'],
325
+ additionalSections: ['model.low_vram'],
326
  accuracyRecoveryAdapters: {
327
  '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors',
328
  },
ui/src/components/HFJobsWorkflow.tsx CHANGED
@@ -24,7 +24,12 @@ type DatasetUploadPlan = {
24
 
25
  const ensurePosixPath = (value: string) => value.replace(/\\/g, '/').replace(/^\/+/, '');
26
 
27
- const INSTRUCTION_ARCHES = new Set(['flux_kontext', 'hidream_e1', 'qwen_image_edit']);
 
 
 
 
 
28
 
29
  const buildDatasetUploadPlan = (jobConfig: JobConfig): DatasetUploadPlan => {
30
  const datasetEntries = jobConfig?.config?.process?.[0]?.datasets ?? [];
 
24
 
25
  const ensurePosixPath = (value: string) => value.replace(/\\/g, '/').replace(/^\/+/, '');
26
 
27
+ const INSTRUCTION_ARCHES = new Set([
28
+ 'flux_kontext',
29
+ 'hidream_e1',
30
+ 'qwen_image_edit',
31
+ 'qwen_image_edit_plus',
32
+ ]);
33
 
34
  const buildDatasetUploadPlan = (jobConfig: JobConfig): DatasetUploadPlan => {
35
  const datasetEntries = jobConfig?.config?.process?.[0]?.datasets ?? [];