sayakpaul

/

flux.1-dev-nf4-with-bnb-integration

like 11

File "D:\code\flux.1-dev-nf4-with-bnb-integration\go.py", line 6, in
model_nf4 = FluxTransformer2DModel.from_pretrained(nf4_id, torch_dtype=torch.bfloat16)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "f:.conda\env\c215\Lib\site-packages\huggingface_hub\utils_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "f:.conda\env\c215\Lib\site-packages\diffusers\models\modeling_utils.py", line 747, in from_pretrained
unexpected_keys = load_model_dict_into_meta(
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "f:.conda\env\c215\Lib\site-packages\diffusers\models\model_loading_utils.py", line 154, in load_model_dict_into_meta
raise ValueError(
ValueError: Cannot load sayakpaul/flux.1-dev-nf4-with-bnb-integration because context_embedder.weight expected shape tensor(..., device='meta', size=(3072, 4096)), but got torch.Size([6291456, 1]). If you want to instead overwrite randomly initialized weights, please make sure to pass both low_cpu_mem_usage=False and ignore_mismatched_sizes=True. For more information, see also: https://github.com/huggingface/diffusers/issues/1619#issuecomment-1345604389 as an example.