-
Notifications
You must be signed in to change notification settings - Fork 362
Description
I am trying to finetune OmniGen on custom dataset using the following code:
accelerate launch --num_processes=1 --use_fsdp --fsdp_offload_params false --fsdp_sharding_strategy SHARD_GRAD_OP --fsdp_auto_wrap_policy TRANSFORMER_BASED_WRAP --fsdp_transformer_layer_cls_to_wrap Phi3DecoderLayer --fsdp_state_dict_type FULL_STATE_DICT --fsdp_forward_prefetch false --fsdp_use_orig_params True --fsdp_cpu_ram_efficient_loading false --fsdp_sync_module_states True train.py --model_name_or_path Shitao/OmniGen-v1-diffusers --json_file /home/Dataset/metadata.jsonl --image_path /home/Dataset --batch_size_per_device 2 --lr 2e-5 --use_lora --lora_rank 8 --max_input_length_limit 18000 --keep_raw_resolution --max_image_size 1024 --gradient_accumulation_steps 1 --ckpt_every 10000 --epochs 100 --log_every 1000 --results_dir omnigen_finetuned_lora
and keep getting the following error:
FileNotFoundError: [Errno 2] No such file or directory: '/home/.cache/huggingface/hub/models--Shitao--OmniGen-v1-diffusers/snapshots/016e2f61d12a98303f6bbdf122687694d7984268/model.pt'
Can someone please help resolving this?