Update model.py
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user