OSError: Error no file named config.json found in directory FLUX.1-dev/transformer

[nikhilcss97]

Hi team,

Getting this error while fine tuning ostris/flux-dev-lora-trainer. Kindly help us with this since it is a blocker.

#############################################
# Running job: flux_train_replicate
#############################################
Running  1 process
Loading Flux model
Loading transformer
Traceback (most recent call last):
File "/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/cog/server/worker.py", line 352, in _predict
result = predict(**payload)
File "/src/train.py", line 355, in train
job.run()
File "/src/train.py", line 97, in run
process.run()
File "/src/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1241, in run
self.sd.load_model()
File "/src/ai-toolkit/toolkit/stable_diffusion_model.py", line 488, in load_model
transformer = FluxTransformer2DModel.from_pretrained(
File "/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
File "/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/diffusers/models/modeling_utils.py", line 612, in from_pretrained
config, unused_kwargs, commit_hash = cls.load_config(
File "/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
File "/root/.pyenv/versions/3.10.15/lib/python3.10/site-packages/diffusers/configuration_utils.py", line 373, in load_config
raise EnvironmentError(
OSError: Error no file named config.json found in directory FLUX.1-dev/transformer.

cc @mattt