Update infer/modules/vc/modules.py
Browse files
infer/modules/vc/modules.py
CHANGED
@@ -101,8 +101,7 @@ class VC:
|
|
101 |
"",
|
102 |
"",
|
103 |
)
|
104 |
-
|
105 |
-
person = sid
|
106 |
logger.info(f"Loading: {person}")
|
107 |
|
108 |
self.cpt = torch.load(person, map_location="cpu")
|
|
|
101 |
"",
|
102 |
"",
|
103 |
)
|
104 |
+
person = f'{os.getenv("weight_root")}/{sid}'
|
|
|
105 |
logger.info(f"Loading: {person}")
|
106 |
|
107 |
self.cpt = torch.load(person, map_location="cpu")
|