Closed bqhuyy closed 1 month ago
Did/Could you try with the latest stable and Nightly PyTorch versions?
@tolgacangoz I'm using the latest one: pytorch 2.4.0
and torchvision 0.19.0
What happens with torch_dtype=torch.bfloat16
or torch_dtype=torch.float32
? Do choosing torch_dtype=torch.float32
and/or removing pipe.enable_attention_slicing()
exceed your memory?
Sadly I can't test with a Mac right now, the pipeline is the same as SDXL, so probably what's generating the problem is the text encoder since it uses custom code.
Maybe give it a try with the original repo but my guess is that you will have the same problem, if it is, that's not something we can fix right now, and maybe you can ask in to the original authors or wait until it's properly integrated into transformers.
@tolgacangoz It seems that the problem comes from pipe.enable_attention_slicing()
. It works after removing it.
Describe the bug
Running Kolors with MPS, the returned result is always black image.
Reproduction
Run this source code. Output image returns black image
Logs
No response
System Info
Who can help?
@asomoza @yiyixuxu