estellea commited on
Commit
cba6223
1 Parent(s): c41321f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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
- pipe_ldm3d = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d")
41
- pipe_ldm3d.to("cuda")
42
 
43
 
44
  prompt ="A picture of some lemons on a table"
45
  name = "lemons"
46
 
47
- output = pipe_ldm3d(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")
 
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")