Spaces:
Sleeping
Sleeping
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")
|