choephix commited on
Commit
2aa7f7d
·
1 Parent(s): a8a74ee

Update app defaults

Browse files
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="1024")
566
  seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
567
- randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
568
- decimation_target = gr.Slider(100000, 500000, label="Decimation Target", value=300000, step=10000)
569
- texture_size = gr.Slider(1024, 4096, label="Texture Size", value=2048, step=1024)
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=12, 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=12, 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=12, 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):
 
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 = 42,
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 = {},