Spaces:

Wuvin
/

Unique3D

Running on Zero

Wuvin commited on May 31

Commit

9536a29

•

1 Parent(s): 65fd5c5

...

Files changed (4) hide show

gradio_app/custom_models/{image2mvimage-objaverseFrot-wonder3d.yaml → image2mvimage.yaml} RENAMED Viewed

File without changes

gradio_app/custom_models/{image2image-objaverseF-rgb2normal.yaml → image2normal.yaml} RENAMED Viewed

File without changes

gradio_app/custom_models/mvimg_prediction.py CHANGED Viewed

@@ -9,7 +9,7 @@ from gradio_app.custom_models.utils import load_pipeline
 from scripts.all_typing import *
 from scripts.utils import session, simple_preprocess
-training_config = "app/custom_models/image2mvimage-objaverseFrot-wonder3d.yaml"
 checkpoint_path = "ckpt/img2mvimg/unet_state_dict.pth"
 trainer, pipeline = load_pipeline(training_config, checkpoint_path)
 pipeline.enable_model_cpu_offload()

 from scripts.all_typing import *
 from scripts.utils import session, simple_preprocess
+training_config = "app/custom_models/image2mvimage.yaml"
 checkpoint_path = "ckpt/img2mvimg/unet_state_dict.pth"
 trainer, pipeline = load_pipeline(training_config, checkpoint_path)
 pipeline.enable_model_cpu_offload()

gradio_app/custom_models/normal_prediction.py CHANGED Viewed

@@ -5,7 +5,7 @@ from gradio_app.custom_models.utils import load_pipeline
 from scripts.utils import rotate_normals_torch
 from scripts.all_typing import *
-training_config = "app/custom_models/image2image-objaverseF-rgb2normal.yaml"
 checkpoint_path = "ckpt/image2normal/unet_state_dict.pth"
 trainer, pipeline = load_pipeline(training_config, checkpoint_path)
 pipeline.enable_model_cpu_offload()

 from scripts.utils import rotate_normals_torch
 from scripts.all_typing import *
+training_config = "app/custom_models/image2normal.yaml"
 checkpoint_path = "ckpt/image2normal/unet_state_dict.pth"
 trainer, pipeline = load_pipeline(training_config, checkpoint_path)
 pipeline.enable_model_cpu_offload()