Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Spaces:
Plachta
/
VALL-E-X
like
332
Running
on
A10G
App
Files
Files
Community
2
9252dce
VALL-E-X
/
utils
/
__init__.py
Plachta
initial commit
b1e1a76
about 1 year ago
raw
Copy download link
history
blame
No virus
198 Bytes
import
torch
import
torch.nn
as
nn
class
Transpose
(nn.Identity):
"""(N, T, D) -> (N, D, T)"""
def
forward
(
self,
input
: torch.Tensor
) -> torch.Tensor:
return
input
.transpose(
1
,
2
)