Spaces:
Sleeping
Sleeping
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 =
|
| 20 |
-
checkpoint_M: str =
|
| 21 |
-
checkpoint_G: str =
|
| 22 |
-
checkpoint_W: str =
|
| 23 |
-
checkpoint_S: str =
|
| 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
|