search

yym0329 / diffusion-project

1 stars 0 forks source link

뭔가 shape 문제 해결.. #7

Closed yym0329 closed 4 days ago

yym0329 commented 6 days ago 
Traceback (most recent call last):
  File "/root/project/DiLightNet/train/train_controlnet.py", line 1273, in <module>
    main(args)
  File "/root/project/DiLightNet/train/train_controlnet.py", line 1139, in main
    down_block_res_samples, mid_block_res_sample = controlnet(
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
    return forward_call(*args, **kwargs)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/diffusers/models/controlnet.py", line 800, in forward
    controlnet_cond = self.controlnet_cond_embedding(controlnet_cond)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
    return forward_call(*args, **kwargs)
  File "/root/.cache/huggingface/modules/diffusers_modules/local/dilightnet--model_helpers/a5ccd0f14a7ff21702351dd806181bedc66d87ad/neuraltexture_controlnet.py", line 125, in forward
    conditioning, shading_hint = torch.split(
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/functional.py", line 207, in split
    return tensor.split(split_size_or_sections, dim)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/_tensor.py", line 983, in split
    return torch._VF.split_with_sizes(self, split_size, dim)
RuntimeError: split_with_sizes expects split_sizes to sum exactly to 15 (input tensor's size at dimension 1), but got split_sizes=[4, 12]
Traceback (most recent call last):
  File "/root/project/DiLightNet/train/train_controlnet.py", line 1273, in <module>
    main(args)
  File "/root/project/DiLightNet/train/train_controlnet.py", line 1139, in main
    down_block_res_samples, mid_block_res_sample = controlnet(
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
    return forward_call(*args, **kwargs)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/diffusers/models/controlnet.py", line 800, in forward
    controlnet_cond = self.controlnet_cond_embedding(controlnet_cond)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
    return forward_call(*args, **kwargs)
  File "/root/.cache/huggingface/modules/diffusers_modules/local/dilightnet--model_helpers/a5ccd0f14a7ff21702351dd806181bedc66d87ad/neuraltexture_controlnet.py", line 125, in forward
    conditioning, shading_hint = torch.split(
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/functional.py", line 207, in split
    return tensor.split(split_size_or_sections, dim)
  File "/root/miniconda3/envs/train/lib/python3.10/site-packages/torch/_tensor.py", line 983, in split
    return torch._VF.split_with_sizes(self, split_size, dim)
RuntimeError: split_with_sizes expects split_sizes to sum exactly to 15 (input tensor's size at dimension 1), but got split_sizes=[4, 12]