Spaces:
Running
on
Zero
Running
on
Zero
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
|