Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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: ['
|
| 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([
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 ?? [];
|