Spaces:

Wuvin
/

Unique3D

Running on Zero

Wuvin commited on May 31

Commit

f38a22d

•

1 Parent(s): 9536a29

rename

Files changed (2) hide show

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.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 = "gradio_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/image2normal.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 = "gradio_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()