runtime error

Exit code: 1. Reason: 1/7 [00:00<00:00, 14.52it/s] Traceback (most recent call last): File "/app/app.py", line 5, in <module> pipe = FluxPipeline.from_pretrained( ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^ "black-forest-labs/FLUX.1-dev", ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ torch_dtype=torch.float8_e4m3fn, ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ device_map="balanced" ^^^^^^^^^^^^^^^^^^^^^ ).to("cuda") ^ File "/usr/local/lib/python3.13/site-packages/huggingface_hub/utils/_validators.py", line 88, in _inner_fn return fn(*args, **kwargs) File "/usr/local/lib/python3.13/site-packages/diffusers/pipelines/pipeline_utils.py", line 1057, in from_pretrained loaded_sub_model = load_sub_model( library_name=library_name, ...<24 lines>... trust_remote_code=trust_remote_code, ) File "/usr/local/lib/python3.13/site-packages/diffusers/pipelines/pipeline_loading_utils.py", line 910, in load_sub_model loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs) File "/usr/local/lib/python3.13/site-packages/transformers/modeling_utils.py", line 4282, in from_pretrained with ContextManagers(model_init_context): ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/transformers/utils/generic.py", line 581, in __enter__ self.stack.enter_context(context_manager) ~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/contextlib.py", line 530, in enter_context result = _enter(cm) File "/usr/local/lib/python3.13/contextlib.py", line 141, in __enter__ return next(self.gen) File "/usr/local/lib/python3.13/site-packages/transformers/modeling_utils.py", line 253, in local_torch_dtype torch.set_default_dtype(dtype) ~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/__init__.py", line 1384, in set_default_dtype _C._set_default_dtype(d) ~~~~~~~~~~~~~~~~~~~~~^^^ TypeError: couldn't find storage object Float8_e4m3fnStorage

Container logs: