Spaces:
Runtime error
Runtime error
from diffusers import StableDiffusionLDM3DPipeline | |
pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d") | |
pipe.to("cuda") | |
prompt ="A picture of some lemons on a table" | |
name = "lemons" | |
output = pipe(prompt) | |
rgb_image, depth_image = output.rgb, output.depth | |
rgb_image[0].save(name+"_ldm3d_rgb.jpg") | |
depth_image[0].save(name+"_ldm3d_depth.png") |