Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,14 @@ pipe.to("cuda")
|
|
47 |
prompt ="360 view of a large bedroom"
|
48 |
name = "bedroom_pano"
|
49 |
|
50 |
-
output = pipe(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
rgb_image, depth_image = output.rgb, output.depth
|
52 |
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
|
53 |
depth_image[0].save(name+"_ldm3d_depth.png")
|
|
|
47 |
prompt ="360 view of a large bedroom"
|
48 |
name = "bedroom_pano"
|
49 |
|
50 |
+
output = pipe(
|
51 |
+
prompt,
|
52 |
+
width=1024,
|
53 |
+
height=512,
|
54 |
+
guidance_scale=7.0,
|
55 |
+
num_inference_steps=50,
|
56 |
+
)
|
57 |
+
|
58 |
rgb_image, depth_image = output.rgb, output.depth
|
59 |
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
|
60 |
depth_image[0].save(name+"_ldm3d_depth.png")
|