add more files

facebookresearch · Jul 27, 2023 · 7f92f5d · 7f92f5d
1 parent 98a9381
commit 7f92f5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/modules/layers/test_transformer.py → tests/models/flava/test_transformer.py b/tests/modules/layers/test_transformer.py → tests/models/flava/test_transformer.py
@@ -9,7 +9,7 @@
 import torch
 from tests.test_utils import assert_expected, set_rng_seed
 from torch import nn
-from torchmultimodal.modules.layers.transformer import (
+from torchmultimodal.models.flava.transformer import (
  TransformerEncoder,
  TransformerEncoderLayer,
 )