Add files using upload-large-folder tool

config.yaml

@@ -0,0 +1,117 @@
+job: extension
+config:
+  name: qwen_2509_object_removal_new
+  process:
+  - type: diffusion_trainer
+    training_folder: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/output
+    sqlite_db_path: ./aitk_db.db
+    device: cuda
+    trigger_word: null
+    performance_log_every: 10
+    network:
+      type: lora
+      linear: 16
+      linear_alpha: 16
+      conv: 16
+      conv_alpha: 16
+      lokr_full_rank: true
+      lokr_factor: -1
+      network_kwargs:
+        ignore_if_contains: []
+    save:
+      dtype: bf16
+      save_every: 50
+      max_step_saves_to_keep: 4
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_train/target
+      mask_path: null
+      mask_min_value: 0.1
+      default_caption: Remove Object
+      caption_ext: txt
+      caption_dropout_rate: 0.05
+      cache_latents_to_disk: true
+      is_reg: false
+      network_weight: 1
+      resolution:
+      - 512
+      controls: []
+      shrink_video_to_frames: true
+      num_frames: 1
+      do_i2v: true
+      flip_x: false
+      flip_y: false
+      control_path_1: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_train/mask_on_background
+      control_path_2: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_train/mask
+      control_path_3: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_train/mae_output
+    train:
+      batch_size: 6
+      bypass_guidance_embedding: false
+      steps: 5500
+      gradient_accumulation: 1
+      train_unet: true
+      train_text_encoder: false
+      gradient_checkpointing: true
+      noise_scheduler: flowmatch
+      optimizer: adamw
+      timestep_type: weighted
+      content_or_style: balanced
+      optimizer_params:
+        weight_decay: 0.0001
+      unload_text_encoder: false
+      cache_text_embeddings: true
+      lr: 0.0005
+      ema_config:
+        use_ema: false
+        ema_decay: 0.99
+      skip_first_sample: true
+      force_first_sample: false
+      disable_sampling: false
+      dtype: bf16
+      diff_output_preservation: false
+      diff_output_preservation_multiplier: 1
+      diff_output_preservation_class: person
+      switch_boundary_every: 1
+      loss_type: mse
+    model:
+      name_or_path: Qwen/Qwen-Image-Edit-2509
+      quantize: true
+      qtype: uint3|ostris/accuracy_recovery_adapters/qwen_image_edit_2509_torchao_uint3.safetensors
+      quantize_te: true
+      qtype_te: uint4
+      arch: qwen_image_edit_plus
+      low_vram: true
+      model_kwargs:
+        match_target_res: false
+      layer_offloading: true
+      layer_offloading_text_encoder_percent: 0
+      layer_offloading_transformer_percent: 1
+    sample:
+      sampler: flowmatch
+      sample_every: 50
+      width: 512
+      height: 512
+      samples:
+      - prompt: Remove Object
+        ctrl_img_1: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mask_on_background/17.jpg
+        ctrl_img_2: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mask/17.jpg
+        ctrl_img_3: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mae_output/17.jpg
+      - prompt: Remove Object
+        ctrl_img_1: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mask_on_background/45.jpg
+        ctrl_img_2: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mask/45.jpg
+        ctrl_img_3: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mae_output/45.jpg
+      - prompt: Remove Object
+        ctrl_img_1: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mask_on_background/49.jpg
+        ctrl_img_2: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mask/49.jpg
+        ctrl_img_3: /home/nguyenvanthanhdat1810@gmail.com/ai-toolkit/datasets/object_removal_val/mae_output/49.jpg
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 4
+      sample_steps: 10
+      num_frames: 1
+      fps: 1
+meta:
+  name: qwen_2509_object_removal_new
+  version: '1.0'

loss_history.json

@@ -0,0 +1,207 @@
+{
+    "step": 650,
+    "epoch": 0,
+    "loss_history": [
+        0.051893532276153564,
+        0.04191002994775772,
+        0.03563995659351349,
+        0.05188479274511337,
+        0.07136755436658859,
+        0.0493043027818203,
+        0.06096412241458893,
+        0.05343471094965935,
+        0.05279016122221947,
+        0.05165542662143707,
+        0.05911332368850708,
+        0.046261124312877655,
+        0.05145987123250961,
+        0.07348300516605377,
+        0.06282848119735718,
+        0.04710949584841728,
+        0.07371529191732407,
+        0.04425327107310295,
+        0.05758678540587425,
+        0.03565998002886772,
+        0.054090213030576706,
+        0.02724049612879753,
+        0.05941595137119293,
+        0.06052125617861748,
+        0.033973388373851776,
+        0.02366490662097931,
+        0.03215572610497475,
+        0.056784581393003464,
+        0.054403841495513916,
+        0.031740397214889526,
+        0.048682890832424164,
+        0.0359157919883728,
+        0.03429361432790756,
+        0.05047951638698578,
+        0.04810015484690666,
+        0.05415612459182739,
+        0.045266665518283844,
+        0.07389190793037415,
+        0.07689608633518219,
+        0.01617683656513691,
+        0.027895715087652206,
+        0.04273153468966484,
+        0.07595288753509521,
+        0.04460812360048294,
+        0.06804212182760239,
+        0.043365802615880966,
+        0.06288819760084152,
+        0.053755879402160645,
+        0.04600618779659271,
+        0.04654350131750107,
+        0.04472731798887253,
+        0.04098866134881973,
+        0.05165230855345726,
+        0.046722885221242905,
+        0.061621394008398056,
+        0.06209284067153931,
+        0.028675343841314316,
+        0.02694562077522278,
+        0.04801797494292259,
+        0.07362907379865646,
+        0.032962650060653687,
+        0.06702133268117905,
+        0.040867991745471954,
+        0.09054657071828842,
+        0.07557031512260437,
+        0.05606820434331894,
+        0.05454063415527344,
+        0.05563917011022568,
+        0.031462959945201874,
+        0.026764918118715286,
+        0.07913929224014282,
+        0.03800187259912491,
+        0.06417255848646164,
+        0.03501094505190849,
+        0.019893253222107887,
+        0.07155022770166397,
+        0.07499650865793228,
+        0.08027419447898865,
+        0.05345085263252258,
+        0.021373271942138672,
+        0.059831492602825165,
+        0.05794444680213928,
+        0.028789382427930832,
+        0.05070403218269348,
+        0.10934846848249435,
+        0.059825070202350616,
+        0.035672876983881,
+        0.03681573644280434,
+        0.05354110151529312,
+        0.03043556772172451,
+        0.05011315643787384,
+        0.03349003195762634,
+        0.06471982598304749,
+        0.03476070240139961,
+        0.059753965586423874,
+        0.05377107858657837,
+        0.047112029045820236,
+        0.05747378617525101,
+        0.07838261127471924,
+        0.04439996927976608,
+        0.051719676703214645,
+        0.07081422954797745,
+        0.04764055833220482,
+        0.06472727656364441,
+        0.025721382349729538,
+        0.04938185214996338,
+        0.0208204947412014,
+        0.04679484665393829,
+        0.053672343492507935,
+        0.04590842127799988,
+        0.02758799120783806,
+        0.05496695637702942,
+        0.052534304559230804,
+        0.042658112943172455,
+        0.04993772506713867,
+        0.04448267072439194,
+        0.021319519728422165,
+        0.029168587177991867,
+        0.037295062094926834,
+        0.04790613800287247,
+        0.058183394372463226,
+        0.03067963197827339,
+        0.05868379399180412,
+        0.050936199724674225,
+        0.04696328565478325,
+        0.029223185032606125,
+        0.053025759756565094,
+        0.04058123379945755,
+        0.06434519588947296,
+        0.042083121836185455,
+        0.043149348348379135,
+        0.047702908515930176,
+        0.044372182339429855,
+        0.030883528292179108,
+        0.029037751257419586,
+        0.015058952383697033,
+        0.03257162496447563,
+        0.025359218940138817,
+        0.036260537803173065,
+        0.05121390148997307,
+        0.05242888256907463,
+        0.02808375284075737,
+        0.07048864662647247,
+        0.061022497713565826,
+        0.03878414258360863,
+        0.042936887592077255,
+        0.05563678964972496,
+        0.02945074997842312,
+        0.020809678360819817,
+        0.05034352093935013,
+        0.06387125700712204,
+        0.028472203761339188,
+        0.05118788033723831,
+        0.04340597987174988,
+        0.07120423763990402,
+        0.044768076390028,
+        0.06425879895687103,
+        0.050438642501831055,
+        0.047228530049324036,
+        0.0628366768360138,
+        0.060967423021793365,
+        0.06019226461648941,
+        0.07305523753166199,
+        0.046951111406087875,
+        0.049580540508031845,
+        0.03337812051177025,
+        0.08303362131118774,
+        0.06565201282501221,
+        0.02152317762374878,
+        0.04263696074485779,
+        0.033167771995067596,
+        0.050332438200712204,
+        0.046692442148923874,
+        0.05139806866645813,
+        0.06747598946094513,
+        0.07169119268655777,
+        0.05237269029021263,
+        0.0478736087679863,
+        0.03808905929327011,
+        0.050925932824611664,
+        0.06277015060186386,
+        0.0564577691257,
+        0.05538750812411308,
+        0.04669160395860672,
+        0.04851044714450836,
+        0.08011412620544434,
+        0.028662553057074547,
+        0.06408864259719849,
+        0.06013261526823044,
+        0.04410611838102341,
+        0.017821576446294785,
+        0.06472691148519516,
+        0.026214828714728355,
+        0.054857559502124786,
+        0.04198151081800461,
+        0.09210877120494843,
+        0.0239681676030159,
+        0.022113261744379997,
+        0.051053278148174286,
+        0.051333315670490265,
+        0.04088010638952255
+    ]
+}

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2314e57b3c2a55f75600e10fbe4ece9db00372e887fdba00d10913a7b831baa2
+size 1176366704

qwen_2509_object_removal_new_000000500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6585a769ce8e9b04acde8c9d6d333a8aec96826d9a6540cc1edf1122f3512ee4
+size 295147304

qwen_2509_object_removal_new_000000550.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d92511fa48f4eda44f7051025c7595cf87ec452ba566c9b910994fa26131a3b0
+size 295148368

qwen_2509_object_removal_new_000000600.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f444924ce4e9807b2957919001cd01723afa4198f58ea6ceba8f7e2679ad43c5
+size 295149440

qwen_2509_object_removal_new_000000650.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06f0f5634d7de807b88defe8f9c721436999a4bc93ab4bd4ca50999a1f2eb7d8
+size 295150496