Update README.md
Browse files
README.md
CHANGED
@@ -37,14 +37,14 @@ Here is how to use this model to get the features of a given text in PyTorch:
|
|
37 |
|
38 |
from diffusers import StableDiffusionLDM3DPipeline
|
39 |
|
40 |
-
|
41 |
-
|
42 |
|
43 |
|
44 |
prompt ="A picture of some lemons on a table"
|
45 |
name = "lemons"
|
46 |
|
47 |
-
output =
|
48 |
rgb_image, depth_image = output.rgb, output.depth
|
49 |
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
|
50 |
depth_image[0].save(name+"_ldm3d_depth.png")
|
|
|
37 |
|
38 |
from diffusers import StableDiffusionLDM3DPipeline
|
39 |
|
40 |
+
pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d")
|
41 |
+
pipe.to("cuda")
|
42 |
|
43 |
|
44 |
prompt ="A picture of some lemons on a table"
|
45 |
name = "lemons"
|
46 |
|
47 |
+
output = pipe(prompt)
|
48 |
rgb_image, depth_image = output.rgb, output.depth
|
49 |
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
|
50 |
depth_image[0].save(name+"_ldm3d_depth.png")
|