search

SUDO-AI-3D / zero123plus

Code repository for Zero123++: a Single Image to Consistent Multi-view Diffusion Base Model.
Apache License 2.0
1.56k stars 108 forks source link

RuntimeError: "LayerNormKernelImpl" not implemented for 'Half' #72

Closed zhxd163 closed 3 months ago

zhxd163 commented 4 months ago

Traceback (most recent call last): File "/zero123plus/venv/lib/python3.11/site-packages/streamlit/runtime/scriptrunner/script_runner.py", line 565, in _run_script exec(code, module.dict) File "/zero123plus/app.py", line 254, in result = pipeline( ^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/root/.cache/huggingface/modules/diffusers_modules/local/pipeline.py", line 360, in call encoded = self.vision_encoder(image_2, output_hidden_states=False) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl return forward_call(args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/transformers/models/clip/modeling_clip.py", line 1304, in forward vision_outputs = self.vision_model( ^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl return forward_call(*args, *kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/transformers/models/clip/modeling_clip.py", line 860, in forward hidden_states = self.pre_layrnorm(hidden_states) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl return self._call_impl(args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/modules/normalization.py", line 201, in forward return F.layer_norm( ^^^^^^^^^^^^^ File "/zero123plus/venv/lib/python3.11/site-packages/torch/nn/functional.py", line 2546, in layer_norm return torch.layer_norm(input, normalized_shape, weight, bias, eps, torch.backends.cudnn.enabled) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: "LayerNormKernelImpl" not implemented for 'Half'

eliphatfs commented 3 months ago

fp16 operations are not fully supported in PyTorch on CPU. You can either specify float32 or use GPU.