Radeon 5700 produces uniformly grey squares
#159
by
eplo
- opened
Why do I get a square of the purest grey instead of any images?
I'm using a Radeon 5700 with 8GB on Ubuntu 22.04. Drivers courtesy of amdgpu-install_5.6.50600-1_all.deb, pytorch courtesy of pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.2
Variables added:
export HSA_OVERRIDE_GFX_VERSION=10.3.0
I ran some model (waifu-diffusion?) once last night but it took ages and was on CPU because I forget the pipe.to("cuda") line.
Deleting torch_dtype=torch.float16 solves this, but introduces other challenges to VRAM conservation which I'm giving the simplest, dumbest, possible solution for here:
from diffusers import StableDiffusionPipeline
import torch
model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id)
pipe = pipe.to("cuda")
prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt, width=400, height=400).images[0]
image.save("astronaut_rides_horse.png")
From here I can play with options to get the image size back up.
eplo
changed discussion status to
closed