JiantaoLin commited on
Commit
4a35d51
·
1 Parent(s): 08b5826
Files changed (1) hide show
  1. pipeline/kiss3d_wrapper.py +2 -3
pipeline/kiss3d_wrapper.py CHANGED
@@ -504,8 +504,8 @@ class kiss3d_wrapper(object):
504
  'num_inference_steps': num_inference_steps,
505
  'guidance_scale': 3.5,
506
  'num_images_per_prompt': 1,
507
- 'width': 2048,
508
- 'height': 1024,
509
  'output_type': 'np',
510
  'generator': generator,
511
  'joint_attention_kwargs': {"scale": lora_scale}
@@ -541,7 +541,6 @@ class kiss3d_wrapper(object):
541
 
542
  return gen_3d_bundle_image_
543
 
544
- @spaces.GPU
545
  def reconstruct_3d_bundle_image(self,
546
  image,
547
  lrm_render_radius=4.15,
 
504
  'num_inference_steps': num_inference_steps,
505
  'guidance_scale': 3.5,
506
  'num_images_per_prompt': 1,
507
+ 'width': 2048//2,
508
+ 'height': 1024//2,
509
  'output_type': 'np',
510
  'generator': generator,
511
  'joint_attention_kwargs': {"scale": lora_scale}
 
541
 
542
  return gen_3d_bundle_image_
543
 
 
544
  def reconstruct_3d_bundle_image(self,
545
  image,
546
  lrm_render_radius=4.15,