"RuntimeError: expected scalar type Half but found Float " when using fp16 #13
by
qpoisson
- opened
pipe = StableDiffusionInpaintPipeline.from_pretrained(
"runwayml/stable-diffusion-inpainting",
use_auth_token = TOKEN,
revision="fp16",
torch_dtype=torch.float16
).to('cuda')
(Loaded to GPU memory successfully.)
Using fp16 as above caused : "RuntimeError: expected scalar type Half but found Float "
at
python3.10/site-packages/transformers/models/clip/modeling_clip.py", line 259, in forward
attn_output = torch.bmm(attn_probs, value_states)
Hey, I had the same problem. What solved it for me was installing the requirements they had used in the notebook:
pip install -qq -U diffusers==0.6.0 transformers ftfy gradio
pip install git+https://github.com/huggingface/diffusers.git
This comment has been hidden
qpoisson
changed discussion status to
closed
qpoisson
changed discussion status to
open