search

NickLucche / stable-diffusion-nvidia-docker

GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web interface. Includes multi-GPUs support.
MIT License
357 stars 43 forks source link

Black image output #11

Open lonmiller opened 2 years ago

lonmiller commented 2 years ago

Output on 6GB VRAM card is just an all-black image at 512x512 and 384x384.

NickLucche commented 2 years ago

Is the output fine at 256x256? If so, you probably don't have enough VRam to run the model at higher resolution; if you have multiple gpus, I am currently working on adding additional multi-gpu support. Alternatively, I believe a lighter model will soon pop up in the open community and I could add that.