From a91be1870f7fcebcb7b08535245364c8ffb22cfb Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Sun, 11 Aug 2024 16:22:43 +0800 Subject: [PATCH] Update model.py --- custom_net/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/custom_net/model.py b/custom_net/model.py index be93a7f..7ad0482 100644 --- a/custom_net/model.py +++ b/custom_net/model.py @@ -5,7 +5,7 @@ import torch.nn as nn import numpy as np from einops import rearrange -from customnet_util import instantiate_from_config +from .customnet_util import instantiate_from_config from .attention import LinearAttention