mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-12 11:32:19 +00:00
updated
This commit is contained in:
parent
9c2a18493a
commit
e06ce7c63a
@ -7,7 +7,7 @@ import torch.nn.functional as F
|
||||
from conformer import ConformerBlock
|
||||
from diffusers.models.activations import get_activation
|
||||
from einops import pack, rearrange, repeat
|
||||
from matcha.models.components.transformer import BasicTransformerBlock
|
||||
from models.components.transformer import BasicTransformerBlock
|
||||
|
||||
|
||||
class SinusoidalPosEmb(torch.nn.Module):
|
||||
|
@ -2,7 +2,7 @@ from abc import ABC
|
||||
|
||||
import torch
|
||||
import torch.nn.functional as F
|
||||
from matcha.models.components.decoder import Decoder
|
||||
from models.components.decoder import Decoder
|
||||
|
||||
|
||||
class BASECFM(torch.nn.Module, ABC):
|
||||
|
@ -5,7 +5,7 @@ import math
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
from einops import rearrange
|
||||
from matcha.model import sequence_mask
|
||||
from model import sequence_mask
|
||||
|
||||
|
||||
class LayerNorm(nn.Module):
|
||||
|
Loading…
x
Reference in New Issue
Block a user