Update model.py

This commit is contained in:
smthemex
2024-08-11 16:22:43 +08:00
committed by GitHub
parent b78d68ccfe
commit a91be1870f
+1 -1
View File
@@ -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