From 72a5bd5f37acf618fe43d64c60dad651f80c55be Mon Sep 17 00:00:00 2001
From: klae01 <101966044+klae01@users.noreply.github.com>
Date: Thu, 8 Sep 2022 10:02:34 +0900
Subject: [PATCH] Update attention.py

---
 pytorchvideo/layers/attention.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorchvideo/layers/attention.py b/pytorchvideo/layers/attention.py
index 5aa4df0..d0f0b80 100644
--- a/pytorchvideo/layers/attention.py
+++ b/pytorchvideo/layers/attention.py
@@ -4,6 +4,7 @@
 
 import numpy
 import torch
+import torch.fx
 import torch.nn as nn
 from torch.nn.common_types import _size_3_t