YinuoGuo27 commited on
Commit
e0cf1e0
·
verified ·
1 Parent(s): a5efe86

Update difpoint/src/config/inference_config.py

Browse files
difpoint/src/config/inference_config.py CHANGED
@@ -16,11 +16,11 @@ from .base_config import PrintableConfig, make_abs_path
16
  class InferenceConfig(PrintableConfig):
17
  # MODEL CONFIG, NOT EXPORTED PARAMS
18
  models_config: str = make_abs_path('./models.yaml') # portrait animation config
19
- checkpoint_F: str = make_abs_path('../../dataset_process/pretrained_weights/liveportrait/base_models/appearance_feature_extractor.pth') # path to checkpoint of F
20
- checkpoint_M: str = make_abs_path('../../dataset_process/pretrained_weights/liveportrait/base_models/motion_extractor.pth') # path to checkpoint pf M
21
- checkpoint_G: str = make_abs_path('../../dataset_process/pretrained_weights/liveportrait/base_models/spade_generator.pth') # path to checkpoint of G
22
- checkpoint_W: str = make_abs_path('../../dataset_process/pretrained_weights/liveportrait/base_models/warping_module.pth') # path to checkpoint of W
23
- checkpoint_S: str = make_abs_path('../../dataset_process/pretrained_weights/liveportrait/retargeting_models/stitching_retargeting_module.pth') # path to checkpoint to S and R_eyes, R_lip
24
 
25
  # EXPORTED PARAMS
26
  flag_use_half_precision: bool = True
 
16
  class InferenceConfig(PrintableConfig):
17
  # MODEL CONFIG, NOT EXPORTED PARAMS
18
  models_config: str = make_abs_path('./models.yaml') # portrait animation config
19
+ checkpoint_F: str = './downloaded_repo/pretrained_weights/liveportrait/base_models/appearance_feature_extractor.pth' # path to checkpoint of F
20
+ checkpoint_M: str = './downloaded_repo/pretrained_weights/liveportrait/base_models/motion_extractor.pth' # path to checkpoint pf M
21
+ checkpoint_G: str = './downloaded_repo/pretrained_weights/liveportrait/base_models/spade_generator.pth' # path to checkpoint of G
22
+ checkpoint_W: str = './downloaded_repo/pretrained_weights/liveportrait/base_models/warping_module.pth' # path to checkpoint of W
23
+ checkpoint_S: str = './downloaded_repo/pretrained_weights/liveportrait/base_models/retargeting_models/stitching_retargeting_module.pth' # path to checkpoint to S and R_eyes, R_lip
24
 
25
  # EXPORTED PARAMS
26
  flag_use_half_precision: bool = True