JiantaoLin
commited on
Commit
·
4a35d51
1
Parent(s):
08b5826
new
Browse files
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,
|