JiantaoLin
commited on
Commit
·
3063c92
1
Parent(s):
5db6228
new
Browse files- app.py +1 -1
- pipeline/kiss3d_wrapper.py +0 -1
- pipeline/pipeline_config/default.yaml +2 -1
app.py
CHANGED
@@ -222,7 +222,7 @@ def bundle_image_to_mesh(
|
|
222 |
lrm_radius = 3.5,
|
223 |
isomer_radius = 4.2,
|
224 |
reconstruction_stage1_steps = 0,
|
225 |
-
reconstruction_stage2_steps =
|
226 |
save_intermediate_results=False,
|
227 |
if_video=True
|
228 |
):
|
|
|
222 |
lrm_radius = 3.5,
|
223 |
isomer_radius = 4.2,
|
224 |
reconstruction_stage1_steps = 0,
|
225 |
+
reconstruction_stage2_steps = 50,
|
226 |
save_intermediate_results=False,
|
227 |
if_video=True
|
228 |
):
|
pipeline/kiss3d_wrapper.py
CHANGED
@@ -616,7 +616,6 @@ def run_text_to_3d(k3d_wrapper,
|
|
616 |
|
617 |
# recon from 3D Bundle image
|
618 |
recon_mesh_path = k3d_wrapper.reconstruct_3d_bundle_image(gen_3d_bundle_image, save_intermediate_results=False)
|
619 |
-
|
620 |
return gen_save_path, recon_mesh_path
|
621 |
|
622 |
def image2mesh_preprocess(k3d_wrapper, input_image_, seed, use_mv_rgb=True):
|
|
|
616 |
|
617 |
# recon from 3D Bundle image
|
618 |
recon_mesh_path = k3d_wrapper.reconstruct_3d_bundle_image(gen_3d_bundle_image, save_intermediate_results=False)
|
|
|
619 |
return gen_save_path, recon_mesh_path
|
620 |
|
621 |
def image2mesh_preprocess(k3d_wrapper, input_image_, seed, use_mv_rgb=True):
|
pipeline/pipeline_config/default.yaml
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
flux:
|
2 |
-
base_model: "https://huggingface.co/Comfy-Org/flux1-dev/blob/main/flux1-dev-fp8.safetensors"
|
|
|
3 |
flux_dtype: 'bf16'
|
4 |
lora: "./checkpoint/flux_lora/rgb_normal_large.safetensors"
|
5 |
controlnet: "InstantX/FLUX.1-dev-Controlnet-Union"
|
|
|
1 |
flux:
|
2 |
+
# base_model: "https://huggingface.co/Comfy-Org/flux1-dev/blob/main/flux1-dev-fp8.safetensors"
|
3 |
+
base_model: "black-forest-labs/FLUX.1-dev"
|
4 |
flux_dtype: 'bf16'
|
5 |
lora: "./checkpoint/flux_lora/rgb_normal_large.safetensors"
|
6 |
controlnet: "InstantX/FLUX.1-dev-Controlnet-Union"
|