Wuvin commited on
Commit
9536a29
1 Parent(s): 65fd5c5
gradio_app/custom_models/{image2mvimage-objaverseFrot-wonder3d.yaml → image2mvimage.yaml} RENAMED
File without changes
gradio_app/custom_models/{image2image-objaverseF-rgb2normal.yaml → image2normal.yaml} RENAMED
File without changes
gradio_app/custom_models/mvimg_prediction.py CHANGED
@@ -9,7 +9,7 @@ from gradio_app.custom_models.utils import load_pipeline
9
  from scripts.all_typing import *
10
  from scripts.utils import session, simple_preprocess
11
 
12
- training_config = "app/custom_models/image2mvimage-objaverseFrot-wonder3d.yaml"
13
  checkpoint_path = "ckpt/img2mvimg/unet_state_dict.pth"
14
  trainer, pipeline = load_pipeline(training_config, checkpoint_path)
15
  pipeline.enable_model_cpu_offload()
 
9
  from scripts.all_typing import *
10
  from scripts.utils import session, simple_preprocess
11
 
12
+ training_config = "app/custom_models/image2mvimage.yaml"
13
  checkpoint_path = "ckpt/img2mvimg/unet_state_dict.pth"
14
  trainer, pipeline = load_pipeline(training_config, checkpoint_path)
15
  pipeline.enable_model_cpu_offload()
gradio_app/custom_models/normal_prediction.py CHANGED
@@ -5,7 +5,7 @@ from gradio_app.custom_models.utils import load_pipeline
5
  from scripts.utils import rotate_normals_torch
6
  from scripts.all_typing import *
7
 
8
- training_config = "app/custom_models/image2image-objaverseF-rgb2normal.yaml"
9
  checkpoint_path = "ckpt/image2normal/unet_state_dict.pth"
10
  trainer, pipeline = load_pipeline(training_config, checkpoint_path)
11
  pipeline.enable_model_cpu_offload()
 
5
  from scripts.utils import rotate_normals_torch
6
  from scripts.all_typing import *
7
 
8
+ training_config = "app/custom_models/image2normal.yaml"
9
  checkpoint_path = "ckpt/image2normal/unet_state_dict.pth"
10
  trainer, pipeline = load_pipeline(training_config, checkpoint_path)
11
  pipeline.enable_model_cpu_offload()