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
|
||||
import torch
|
||||
import os
|
||||
import copy # Add this for deep copying
|
||||
import copy
|
||||
|
||||
|
||||
class AIIA(nn.Module):
|
||||
def __init__(self, config: AIIAConfig, **kwargs):
|
||||
|
|
Loading…
Reference in New Issue