correct copying
This commit is contained in:
parent
32daaadddd
commit
e5a5618160
|
@ -1,8 +1,9 @@
|
||||||
from config import AIIAConfig
|
from .config import AIIAConfig
|
||||||
from torch import nn
|
from torch import nn
|
||||||
import torch
|
import torch
|
||||||
import os
|
import os
|
||||||
import copy # Add this for deep copying
|
import copy
|
||||||
|
|
||||||
|
|
||||||
class AIIA(nn.Module):
|
class AIIA(nn.Module):
|
||||||
def __init__(self, config: AIIAConfig, **kwargs):
|
def __init__(self, config: AIIAConfig, **kwargs):
|
||||||
|
|
Loading…
Reference in New Issue