Skip to content

Cannot find model.pt file while trying to fine tune OmniGen #219

@Reesheetaa

Description

@Reesheetaa

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions