From d49bcecbfd939b0611a9165f3c42b8fcf8830253 Mon Sep 17 00:00:00 2001 From: BBing <67486385@qq.com> Date: Wed, 4 Oct 2023 10:26:52 +0800 Subject: [PATCH] Update embed.py --- models/embed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/embed.py b/models/embed.py index 905c0a5..32797af 100644 --- a/models/embed.py +++ b/models/embed.py @@ -1,4 +1,4 @@ - +import import torch import torch.nn as nn import torch.nn.functional as F