mirror of
https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git
synced 2025-01-21 14:00:06 +08:00
Update gui.py
This commit is contained in:
parent
fa97c3f8bd
commit
a68037be3c
6
gui.py
6
gui.py
@ -150,8 +150,12 @@ class RVC:
|
|||||||
assert feats.dim() == 1, feats.dim()
|
assert feats.dim() == 1, feats.dim()
|
||||||
feats = feats.view(1, -1)
|
feats = feats.view(1, -1)
|
||||||
padding_mask = torch.BoolTensor(feats.shape).fill_(False)
|
padding_mask = torch.BoolTensor(feats.shape).fill_(False)
|
||||||
|
if Config.is_half:
|
||||||
|
feats = feats.half()
|
||||||
|
else:
|
||||||
|
feats = feats.float()
|
||||||
inputs = {
|
inputs = {
|
||||||
"source": feats.half().to(device),
|
"source": feats.to(device),
|
||||||
"padding_mask": padding_mask.to(device),
|
"padding_mask": padding_mask.to(device),
|
||||||
"output_layer": 9 if self.version == "v1" else 12,
|
"output_layer": 9 if self.version == "v1" else 12,
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user