From 6e1ad65edd7cbb0f8eb7a48991e9f000f554761d Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 5 Jun 2022 21:16:32 +0200
Subject: Rename mlp to ff

Rename mlp to ff
---
 text_recognizer/networks/conformer/block.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'text_recognizer/networks/conformer/block.py')

diff --git a/text_recognizer/networks/conformer/block.py b/text_recognizer/networks/conformer/block.py
index d9782e8..4b31aec 100644
--- a/text_recognizer/networks/conformer/block.py
+++ b/text_recognizer/networks/conformer/block.py
@@ -5,7 +5,7 @@ from typing import Optional
 from torch import nn, Tensor
 from text_recognizer.networks.conformer.conv import ConformerConv
 
-from text_recognizer.networks.conformer.mlp import MLP
+from text_recognizer.networks.conformer.ff import Feedforward
 from text_recognizer.networks.conformer.scale import Scale
 from text_recognizer.networks.transformer.attention import Attention
 from text_recognizer.networks.transformer.norm import PreNorm
@@ -15,7 +15,7 @@ class ConformerBlock(nn.Module):
     def __init__(
         self,
         dim: int,
-        ff: MLP,
+        ff: Feedforward,
         attn: Attention,
         conv: ConformerConv,
     ) -> None:
-- 
cgit v1.2.3-70-g09d2