{"results_dir": "/nvme-data/Komal/documents/results/DREAMBOOT/full/nmf/", "model_name_or_path": "Shitao/OmniGen-v1", "json_file": "/home/mbzuaiser/Documents/Komal/PolicyGen/OmniGen/toy_data/concepts/dreambooth_eval.jsonl", "image_path": "/home/mbzuaiser/Documents/Komal/PolicyGen/OmniGen/toy_data/concepts", "epochs": 2000, "batch_size_per_device": 2, "vae_path": null, "num_workers": 4, "log_every": 1, "ckpt_every": 2000, "max_grad_norm": 1.0, "lr": 0.001, "max_input_length_limit": 18000, "condition_dropout_prob": 0.01, "adam_weight_decay": 0.0, "keep_raw_resolution": true, "max_image_size": 1024, "use_lora": false, "use_para": true, "lora_rank": 8, "use_svd": false, "use_ema": false, "lr_scheduler": "constant", "decomposition_method": "nmf", "lr_warmup_steps": 1000, "report_to": "tensorboard", "mixed_precision": "bf16", "gradient_accumulation_steps": 1}