add elevation in call

pipeline_mvdream.py

@@ -435,6 +435,7 @@ class MVDreamPipeline(DiffusionPipeline):
         image: Optional[np.ndarray] = None,
         height: int = 256,
         width: int = 256,
+        elevation: float = 0,
         num_inference_steps: int = 50,
         guidance_scale: float = 7.0,
         negative_prompt: str = "",
@@ -489,10 +490,8 @@ class MVDreamPipeline(DiffusionPipeline):
             None,
         )
 
-        if image is not None:
-            camera = get_camera(num_frames, elevation=5, extra_view=True).to(dtype=latents.dtype, device=device)
-        else:
-            camera = get_camera(num_frames, elevation=15, extra_view=False).to(dtype=latents.dtype, device=device)
+        # Get camera
+        camera = get_camera(num_frames, elevation=elevation, extra_view=(image is not None)).to(dtype=latents.dtype, device=device)
         camera = camera.repeat_interleave(num_images_per_prompt, dim=0)
 
         # Prepare extra step kwargs.

run_imagedream.py

@@ -20,7 +20,7 @@ args = parser.parse_args()
 
 for i in range(5):
     input_image = kiui.read_image(args.image, mode='float')
-    image = pipe(args.prompt, input_image, guidance_scale=5)
+    image = pipe(args.prompt, input_image, guidance_scale=5, num_inference_steps=30, elevation=0)
     grid = np.concatenate(
         [
             np.concatenate([image[0], image[2]], axis=0),

run_mvdream.py

@@ -19,7 +19,7 @@ parser.add_argument("prompt", type=str, default="a cute owl 3d model")
 args = parser.parse_args()
 
 for i in range(5):
-    image = pipe(args.prompt)
+    image = pipe(args.prompt, guidance_scale=5, num_inference_steps=30, elevation=0)
     grid = np.concatenate(
         [
             np.concatenate([image[0], image[2]], axis=0),