Add files using upload-large-folder tool

config.yaml

@@ -0,0 +1,117 @@
+job: extension
+config:
+  name: qwen_2509_object_removal
+  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/with_object
+      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/with_object/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/with_object/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/with_object/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
+  version: '1.0'

loss_history.json

@@ -0,0 +1,357 @@
+{
+    "step": 350,
+    "epoch": 0,
+    "loss_history": [
+        0.0717553198337555,
+        0.09106086939573288,
+        0.10163074731826782,
+        0.08771749585866928,
+        0.04425919055938721,
+        0.10402189195156097,
+        0.0860668271780014,
+        0.05974728614091873,
+        0.06216242164373398,
+        0.08865997195243835,
+        0.04145634174346924,
+        0.07419457286596298,
+        0.10662155598402023,
+        0.08226007223129272,
+        0.05869744345545769,
+        0.0321805439889431,
+        0.054890941828489304,
+        0.05829688161611557,
+        0.05702868849039078,
+        0.07853753864765167,
+        0.05898887664079666,
+        0.053761959075927734,
+        0.048632144927978516,
+        0.047201674431562424,
+        0.08215396106243134,
+        0.051663175225257874,
+        0.06857939809560776,
+        0.08221539109945297,
+        0.07198384404182434,
+        0.04746587201952934,
+        0.03505876660346985,
+        0.06263525784015656,
+        0.0348798930644989,
+        0.03558044880628586,
+        0.08985966444015503,
+        0.08350600302219391,
+        0.059380173683166504,
+        0.054323844611644745,
+        0.07983306795358658,
+        0.1023111641407013,
+        0.05226708948612213,
+        0.08248939365148544,
+        0.06335695087909698,
+        0.07242630422115326,
+        0.058780692517757416,
+        0.05363859981298447,
+        0.06319660693407059,
+        0.061154574155807495,
+        0.0606878362596035,
+        0.02633911930024624,
+        0.0483890175819397,
+        0.04612479731440544,
+        0.05468442663550377,
+        0.06710803508758545,
+        0.06309658288955688,
+        0.037624292075634,
+        0.07683953642845154,
+        0.05704308673739433,
+        0.07780913263559341,
+        0.0588805191218853,
+        0.07304685562849045,
+        0.045938268303871155,
+        0.05717086046934128,
+        0.06624903529882431,
+        0.03854874521493912,
+        0.04800817370414734,
+        0.05623609945178032,
+        0.05680468678474426,
+        0.07534592598676682,
+        0.059384893625974655,
+        0.05943627282977104,
+        0.06484764069318771,
+        0.054009612649679184,
+        0.06521561741828918,
+        0.07848631590604782,
+        0.058265697211027145,
+        0.07383449375629425,
+        0.04961680620908737,
+        0.09161170572042465,
+        0.057914238423109055,
+        0.06883111596107483,
+        0.041094280779361725,
+        0.04295092821121216,
+        0.044410280883312225,
+        0.08960114419460297,
+        0.05625128373503685,
+        0.04787581413984299,
+        0.08865277469158173,
+        0.05116886645555496,
+        0.05804094672203064,
+        0.04021447151899338,
+        0.05786088854074478,
+        0.03961793705821037,
+        0.05596750229597092,
+        0.06411083042621613,
+        0.0543573722243309,
+        0.04326349124312401,
+        0.028126582503318787,
+        0.04684855788946152,
+        0.04912514612078667,
+        0.0603066086769104,
+        0.06141732260584831,
+        0.022980835288763046,
+        0.09517131745815277,
+        0.042873457074165344,
+        0.04191361367702484,
+        0.03220736235380173,
+        0.036481522023677826,
+        0.029959311708807945,
+        0.05177910253405571,
+        0.0506553053855896,
+        0.09598179906606674,
+        0.05082204192876816,
+        0.04847095161676407,
+        0.07127466797828674,
+        0.05119007080793381,
+        0.06840521097183228,
+        0.058361709117889404,
+        0.03772614523768425,
+        0.029362855479121208,
+        0.050974708050489426,
+        0.04146232455968857,
+        0.06297391653060913,
+        0.07563933730125427,
+        0.05116645619273186,
+        0.07868105918169022,
+        0.03472984954714775,
+        0.06736486405134201,
+        0.07980863004922867,
+        0.03578195348381996,
+        0.056639157235622406,
+        0.0589706227183342,
+        0.0726524144411087,
+        0.07339595258235931,
+        0.07703565806150436,
+        0.05096810683608055,
+        0.03412363678216934,
+        0.025429053232073784,
+        0.06726783514022827,
+        0.03710677847266197,
+        0.05120859667658806,
+        0.028654327616095543,
+        0.04766024276614189,
+        0.05502372235059738,
+        0.07161981612443924,
+        0.06771059334278107,
+        0.04256628081202507,
+        0.04421256110072136,
+        0.050081901252269745,
+        0.0433955155313015,
+        0.046140387654304504,
+        0.06180858612060547,
+        0.040949009358882904,
+        0.0575898215174675,
+        0.04323597252368927,
+        0.04798489063978195,
+        0.049740683287382126,
+        0.05452745780348778,
+        0.0412115640938282,
+        0.06114327162504196,
+        0.03975970298051834,
+        0.057137250900268555,
+        0.0807805210351944,
+        0.05234125256538391,
+        0.083372563123703,
+        0.025964651256799698,
+        0.06275730580091476,
+        0.07823973149061203,
+        0.03659472241997719,
+        0.050935350358486176,
+        0.053869735449552536,
+        0.05415455996990204,
+        0.07607614248991013,
+        0.06579896807670593,
+        0.05942368134856224,
+        0.03718090057373047,
+        0.06675874441862106,
+        0.044610124081373215,
+        0.03264891356229782,
+        0.05212625488638878,
+        0.05661637336015701,
+        0.04439613223075867,
+        0.03446171432733536,
+        0.05344244837760925,
+        0.05158240720629692,
+        0.04657692462205887,
+        0.044233910739421844,
+        0.04421672224998474,
+        0.04825364798307419,
+        0.06608051061630249,
+        0.035017237067222595,
+        0.01604272425174713,
+        0.05559912323951721,
+        0.055130861699581146,
+        0.03835352882742882,
+        0.020352069288492203,
+        0.036496952176094055,
+        0.037345901131629944,
+        0.04420202225446701,
+        0.038464609533548355,
+        0.07009484618902206,
+        0.03797260671854019,
+        0.04699648171663284,
+        0.07374894618988037,
+        0.07532134652137756,
+        0.04370319843292236,
+        0.05667514353990555,
+        0.07589396834373474,
+        0.0782490223646164,
+        0.04563229903578758,
+        0.04424218460917473,
+        0.08060942590236664,
+        0.02128773182630539,
+        0.016241226345300674,
+        0.07119874656200409,
+        0.06271836906671524,
+        0.08548914641141891,
+        0.05263606831431389,
+        0.0619259774684906,
+        0.05057978257536888,
+        0.05586700886487961,
+        0.0664830356836319,
+        0.04831165075302124,
+        0.07262303680181503,
+        0.047449029982089996,
+        0.04055073857307434,
+        0.050719670951366425,
+        0.038560692220926285,
+        0.06396868824958801,
+        0.06640258431434631,
+        0.05889654904603958,
+        0.048576802015304565,
+        0.04657858610153198,
+        0.08124115318059921,
+        0.05805934965610504,
+        0.04692807048559189,
+        0.027358010411262512,
+        0.03842280060052872,
+        0.021573608741164207,
+        0.041345395147800446,
+        0.058279287070035934,
+        0.06811287999153137,
+        0.05025707930326462,
+        0.06645620614290237,
+        0.05002322420477867,
+        0.03902001678943634,
+        0.02735285460948944,
+        0.07645498961210251,
+        0.04375564679503441,
+        0.03034152090549469,
+        0.0621485710144043,
+        0.00728888064622879,
+        0.05927498638629913,
+        0.05112169310450554,
+        0.04102331027388573,
+        0.07570874691009521,
+        0.05224767327308655,
+        0.07497574388980865,
+        0.042466603219509125,
+        0.05495554208755493,
+        0.06564036011695862,
+        0.048475466668605804,
+        0.07831702381372452,
+        0.03851762041449547,
+        0.05344545841217041,
+        0.047441981732845306,
+        0.026691745966672897,
+        0.07037435472011566,
+        0.05719257518649101,
+        0.05371653288602829,
+        0.048148687928915024,
+        0.03023616597056389,
+        0.037204988300800323,
+        0.06274713575839996,
+        0.04337472468614578,
+        0.05810517072677612,
+        0.04830416291952133,
+        0.05416257306933403,
+        0.058113690465688705,
+        0.07381264865398407,
+        0.04838205501437187,
+        0.07907748222351074,
+        0.04346910119056702,
+        0.018346961587667465,
+        0.043756917119026184,
+        0.01935887709259987,
+        0.06304316967725754,
+        0.0747736245393753,
+        0.053791314363479614,
+        0.03450062498450279,
+        0.041505515575408936,
+        0.04262706637382507,
+        0.06708305329084396,
+        0.04674195870757103,
+        0.0632915049791336,
+        0.059526052325963974,
+        0.02764313668012619,
+        0.05154145881533623,
+        0.07235918939113617,
+        0.038563504815101624,
+        0.06667011976242065,
+        0.0663178563117981,
+        0.03310840576887131,
+        0.030033793300390244,
+        0.03238404169678688,
+        0.0657908171415329,
+        0.038437142968177795,
+        0.06293841451406479,
+        0.027537589892745018,
+        0.03871290385723114,
+        0.04816951975226402,
+        0.03599255532026291,
+        0.050419699400663376,
+        0.07281208038330078,
+        0.05480337142944336,
+        0.06078067421913147,
+        0.03726741671562195,
+        0.07435443997383118,
+        0.03836868703365326,
+        0.04493751376867294,
+        0.02936941757798195,
+        0.06643304973840714,
+        0.04007839411497116,
+        0.054177187383174896,
+        0.06291265040636063,
+        0.024266723543405533,
+        0.0387108288705349,
+        0.04670628905296326,
+        0.030854186043143272,
+        0.051181610673666,
+        0.0367170013487339,
+        0.06125319004058838,
+        0.06676872819662094,
+        0.023486964404582977,
+        0.04489211365580559,
+        0.028676077723503113,
+        0.027890313416719437,
+        0.06824913620948792,
+        0.04514992982149124,
+        0.05180990323424339,
+        0.0689506083726883,
+        0.03876955434679985,
+        0.05474601313471794,
+        0.07524611055850983,
+        0.05128498002886772,
+        0.02429131232202053,
+        0.057598214596509933,
+        0.052720166742801666,
+        0.01701473258435726,
+        0.03177155926823616,
+        0.032222338020801544
+    ]
+}

optimizer.pt

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

qwen_2509_object_removal_000000200.safetensors

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

qwen_2509_object_removal_000000250.safetensors

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

qwen_2509_object_removal_000000300.safetensors

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

qwen_2509_object_removal_000000350.safetensors

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