Pytorch>=2.6 fix (#52)
* Fix bug with Intel GPUs. * Updated .gitignore * Fix newer pytorch version's error
This commit is contained in:
@@ -11,8 +11,8 @@ github = "https://github.com/RVC-Project/Retrieval-based-Voice-Conversion"
|
|||||||
[tool.poetry.dependencies]
|
[tool.poetry.dependencies]
|
||||||
python = "^3.10"
|
python = "^3.10"
|
||||||
torch = "^2.1.0"
|
torch = "^2.1.0"
|
||||||
fairseq = "^0.12.2"
|
#fairseq = "^0.12.2"
|
||||||
# fairseq = {git = "https://github.com/Tps-F/fairseq.git", branch="main"}
|
fairseq = {git = "https://github.com/Tps-F/fairseq.git", branch="main"}
|
||||||
soundfile = "^0.12.1"
|
soundfile = "^0.12.1"
|
||||||
librosa = "^0.10.1"
|
librosa = "^0.10.1"
|
||||||
praat-parselmouth = "^0.4.3"
|
praat-parselmouth = "^0.4.3"
|
||||||
|
|||||||
@@ -47,7 +47,7 @@ class VC:
|
|||||||
person = sid if os.path.exists(sid) else f'{os.getenv("weight_root")}/{sid}'
|
person = sid if os.path.exists(sid) else f'{os.getenv("weight_root")}/{sid}'
|
||||||
logger.info(f"Loading: {person}")
|
logger.info(f"Loading: {person}")
|
||||||
|
|
||||||
self.cpt = torch.load(person, map_location="cpu")
|
self.cpt = torch.load(person, weights_only=False, map_location="cpu")
|
||||||
self.tgt_sr = self.cpt["config"][-1]
|
self.tgt_sr = self.cpt["config"][-1]
|
||||||
self.cpt["config"][-3] = self.cpt["weight"]["emb_g.weight"].shape[0] # n_spk
|
self.cpt["config"][-3] = self.cpt["weight"]["emb_g.weight"].shape[0] # n_spk
|
||||||
self.if_f0 = self.cpt.get("f0", 1)
|
self.if_f0 = self.cpt.get("f0", 1)
|
||||||
|
|||||||
Reference in New Issue
Block a user