Update extract_feature_print.py
Browse files- extract_feature_print.py +1 -1
extract_feature_print.py
CHANGED
|
@@ -89,7 +89,7 @@ models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task(
|
|
| 89 |
if Custom_Embed == False:
|
| 90 |
model = models[0]
|
| 91 |
else:
|
| 92 |
-
model = HubertModelWithFinalProj.from_pretrained(
|
| 93 |
model = model.to(device)
|
| 94 |
printt("move model to %s" % device)
|
| 95 |
if device not in ["mps", "cpu"]:
|
|
|
|
| 89 |
if Custom_Embed == False:
|
| 90 |
model = models[0]
|
| 91 |
else:
|
| 92 |
+
model = HubertModelWithFinalProj.from_pretrained("/kaggle/working/Mangio-RVC-Fork/Custom/")
|
| 93 |
model = model.to(device)
|
| 94 |
printt("move model to %s" % device)
|
| 95 |
if device not in ["mps", "cpu"]:
|