Fix incorrect image embedding when running with a single GPU and 24GB VRAM aa5199e verified xdedss commited on Jun 22