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")
|