deepseek-ai / DreamCraft3D

[ICLR 2024] Official implementation of DreamCraft3D: Hierarchical 3D Generation with Bootstrapped Diffusion Prior
https://mrtornado24.github.io/DreamCraft3D/
MIT License
1.96k stars 86 forks source link

_pickle.UnpicklingError: invalid load key, '<'. #61

Open RoenTh opened 3 months ago

RoenTh commented 3 months ago

Hello, thank you for your work! I wanted to ask you, why do I keep getting this error when I try to run DreamCraft3D?

/opt/conda/lib/python3.10/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  torch.utils._pytree._register_pytree_node(
Seed set to 0
[INFO] Loading Deep Floyd ...
/opt/conda/lib/python3.10/site-packages/huggingface_hub/file_download.py:1132: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.
  warnings.warn(

A mixture of fp16 and non-fp16 filenames will be loaded.
Loaded fp16 filenames:
[unet/diffusion_pytorch_model.fp16.safetensors, safety_checker/model.fp16.safetensors, text_encoder/model.fp16-00002-of-00002.safetensors, text_encoder/model.fp16-00001-of-00002.safetensors]
Loaded non-fp16 filenames:
[watermarker/diffusion_pytorch_model.safetensors
If this behavior is not expected, please check your folder structure.
Loading pipeline components...:   0%|                     | 0/3 [00:00<?, ?it/s]You are using the default legacy behaviour of the <class 'transformers.models.t5.tokenization_t5.T5Tokenizer'>. This is expected, and simply means that the `legacy` (previous) behavior will be used so nothing changes for you. If you want to use the new behaviour, set `legacy=False`. This should only be set if you understand what it means, and thoroughly read the reason why this was added as explained in https://github.com/huggingface/transformers/pull/24565
Loading pipeline components...: 100%|█████████████| 3/3 [00:00<00:00, 10.68it/s]
[INFO] Loaded Deep Floyd!
[INFO] Loading Stable Zero123 ...
Traceback (most recent call last):
  File "/DreamCraft3D/launch.py", line 252, in <module>
    main(args, extras)
  File "/DreamCraft3D/launch.py", line 120, in main
    system: BaseSystem = threestudio.find(cfg.system_type)(
  File "/DreamCraft3D/threestudio/systems/base.py", line 45, in __init__
    self.configure()
  File "/DreamCraft3D/threestudio/systems/dreamcraft3d.py", line 45, in configure
    self.guidance_3d = threestudio.find(self.cfg.guidance_3d_type)(
  File "/DreamCraft3D/threestudio/utils/base.py", line 83, in __init__
    self.configure(*args, **kwargs)
  File "/DreamCraft3D/threestudio/models/guidance/stable_zero123_guidance.py", line 105, in configure
    self.model = load_model_from_config(
  File "/DreamCraft3D/threestudio/models/guidance/stable_zero123_guidance.py", line 41, in load_model_from_config
    pl_sd = torch.load(ckpt, map_location="cpu")
  File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 1040, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 1262, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '<'.