Update app defaults
Browse files- app.py +7 -7
- trellis2/pipelines/trellis2_image_to_3d.py +1 -1
app.py
CHANGED
|
@@ -562,11 +562,11 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
|
|
| 562 |
with gr.Column(scale=1, min_width=360):
|
| 563 |
image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=400)
|
| 564 |
|
| 565 |
-
resolution = gr.Radio(["512", "1024", "1536"], label="Resolution", value="
|
| 566 |
seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
|
| 567 |
-
randomize_seed = gr.Checkbox(label="Randomize Seed", value=
|
| 568 |
-
decimation_target = gr.Slider(
|
| 569 |
-
texture_size = gr.Slider(1024, 4096, label="Texture Size", value=
|
| 570 |
|
| 571 |
generate_btn = gr.Button("Generate")
|
| 572 |
|
|
@@ -575,19 +575,19 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
|
|
| 575 |
with gr.Row():
|
| 576 |
ss_guidance_strength = gr.Slider(1.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
|
| 577 |
ss_guidance_rescale = gr.Slider(0.0, 1.0, label="Guidance Rescale", value=0.7, step=0.01)
|
| 578 |
-
ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=
|
| 579 |
ss_rescale_t = gr.Slider(1.0, 6.0, label="Rescale T", value=5.0, step=0.1)
|
| 580 |
gr.Markdown("Stage 2: Shape Generation")
|
| 581 |
with gr.Row():
|
| 582 |
shape_slat_guidance_strength = gr.Slider(1.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
|
| 583 |
shape_slat_guidance_rescale = gr.Slider(0.0, 1.0, label="Guidance Rescale", value=0.5, step=0.01)
|
| 584 |
-
shape_slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=
|
| 585 |
shape_slat_rescale_t = gr.Slider(1.0, 6.0, label="Rescale T", value=3.0, step=0.1)
|
| 586 |
gr.Markdown("Stage 3: Material Generation")
|
| 587 |
with gr.Row():
|
| 588 |
tex_slat_guidance_strength = gr.Slider(1.0, 10.0, label="Guidance Strength", value=1.0, step=0.1)
|
| 589 |
tex_slat_guidance_rescale = gr.Slider(0.0, 1.0, label="Guidance Rescale", value=0.0, step=0.01)
|
| 590 |
-
tex_slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=
|
| 591 |
tex_slat_rescale_t = gr.Slider(1.0, 6.0, label="Rescale T", value=3.0, step=0.1)
|
| 592 |
|
| 593 |
with gr.Column(scale=10):
|
|
|
|
| 562 |
with gr.Column(scale=1, min_width=360):
|
| 563 |
image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=400)
|
| 564 |
|
| 565 |
+
resolution = gr.Radio(["512", "1024", "1536"], label="Resolution", value="512")
|
| 566 |
seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
|
| 567 |
+
randomize_seed = gr.Checkbox(label="Randomize Seed", value=False)
|
| 568 |
+
decimation_target = gr.Slider(20000, 500000, label="Decimation Target", value=20000, step=10000)
|
| 569 |
+
texture_size = gr.Slider(1024, 4096, label="Texture Size", value=1024, step=1024)
|
| 570 |
|
| 571 |
generate_btn = gr.Button("Generate")
|
| 572 |
|
|
|
|
| 575 |
with gr.Row():
|
| 576 |
ss_guidance_strength = gr.Slider(1.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
|
| 577 |
ss_guidance_rescale = gr.Slider(0.0, 1.0, label="Guidance Rescale", value=0.7, step=0.01)
|
| 578 |
+
ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=8, step=1)
|
| 579 |
ss_rescale_t = gr.Slider(1.0, 6.0, label="Rescale T", value=5.0, step=0.1)
|
| 580 |
gr.Markdown("Stage 2: Shape Generation")
|
| 581 |
with gr.Row():
|
| 582 |
shape_slat_guidance_strength = gr.Slider(1.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
|
| 583 |
shape_slat_guidance_rescale = gr.Slider(0.0, 1.0, label="Guidance Rescale", value=0.5, step=0.01)
|
| 584 |
+
shape_slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=8, step=1)
|
| 585 |
shape_slat_rescale_t = gr.Slider(1.0, 6.0, label="Rescale T", value=3.0, step=0.1)
|
| 586 |
gr.Markdown("Stage 3: Material Generation")
|
| 587 |
with gr.Row():
|
| 588 |
tex_slat_guidance_strength = gr.Slider(1.0, 10.0, label="Guidance Strength", value=1.0, step=0.1)
|
| 589 |
tex_slat_guidance_rescale = gr.Slider(0.0, 1.0, label="Guidance Rescale", value=0.0, step=0.01)
|
| 590 |
+
tex_slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=8, step=1)
|
| 591 |
tex_slat_rescale_t = gr.Slider(1.0, 6.0, label="Rescale T", value=3.0, step=0.1)
|
| 592 |
|
| 593 |
with gr.Column(scale=10):
|
trellis2/pipelines/trellis2_image_to_3d.py
CHANGED
|
@@ -483,7 +483,7 @@ class Trellis2ImageTo3DPipeline(Pipeline):
|
|
| 483 |
self,
|
| 484 |
image: Image.Image,
|
| 485 |
num_samples: int = 1,
|
| 486 |
-
seed: int =
|
| 487 |
sparse_structure_sampler_params: dict = {},
|
| 488 |
shape_slat_sampler_params: dict = {},
|
| 489 |
tex_slat_sampler_params: dict = {},
|
|
|
|
| 483 |
self,
|
| 484 |
image: Image.Image,
|
| 485 |
num_samples: int = 1,
|
| 486 |
+
seed: int = 0,
|
| 487 |
sparse_structure_sampler_params: dict = {},
|
| 488 |
shape_slat_sampler_params: dict = {},
|
| 489 |
tex_slat_sampler_params: dict = {},
|