runtime error

Exit code: 1. Reason: fig.json: 100%|██████████| 914/914 [00:00<00:00, 8.79MB/s] vae/diffusion_pytorch_model.safetensors: 0%| | 0.00/168M [00:00<?, ?B/s] vae/diffusion_pytorch_model.safetensors: 100%|██████████| 168M/168M [00:00<00:00, 214MB/s] Loading pipeline components...: 0%| | 0/7 [00:00<?, ?it/s]You set `add_prefix_space`. The tokenizer needs to be converted from the slow tokenizers Loading checkpoint shards: 0%| | 0/3 [00:00<?, ?it/s] Loading checkpoint shards: 67%|██████▋ | 2/3 [00:01<00:00, 1.65it/s] Loading checkpoint shards: 100%|██████████| 3/3 [00:01<00:00, 2.42it/s] Loading pipeline components...: 57%|█████▋ | 4/7 [00:01<00:01, 2.30it/s]`torch_dtype` is deprecated! Use `dtype` instead! Loading pipeline components...: 57%|█████▋ | 4/7 [00:01<00:01, 2.30it/s] Traceback (most recent call last): File "/home/user/app/app.py", line 20, in <module> pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda") File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 1021, in from_pretrained loaded_sub_model = load_sub_model( File "/usr/local/lib/python3.10/site-packages/diffusers/pipelines/pipeline_loading_utils.py", line 857, in load_sub_model loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 277, in _wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4974, in from_pretrained model = cls(config, *model_args, **model_kwargs) TypeError: T5EncoderModel.__init__() got an unexpected keyword argument 'offload_state_dict'

Container logs: