AttributeError: 'DaViT' object has no attribute '_initialize_weights' #7
Description
my error：
SequenceDataset: loading dataset into memory...
100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 200/200 [00:02<00:00, 99.83it/s]
SequenceDataset: caching get_item calls...
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 30154/30154 [00:01<00:00, 21255.86it/s]
SequenceDataset: loading dataset into memory...
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 200/200 [00:00<00:00, 1064.65it/s]
SequenceDataset: caching get_item calls...
100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 30154/30154 [00:00<00:00, 145251.07it/s]
/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/timm/models/layers/init.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers
warnings.warn(f"Importing from {name} is deprecated, please import via timm.layers", FutureWarning)
Traceback (most recent call last):
File "/home/robot/mimic/mimictest/mimictest/Scripts/RobomimicExperiments/FlowFlorencePi0ImageMimic.py", line 138, in
net = FlorencePi0Net(
File "/home/robot/mimic/mimictest/mimictest/Nets/FlorencePi0Net.py", line 31, in init
self.net = AutoModelForCausalLM.from_pretrained(path, trust_remote_code=True,local_files_only=True)
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/models/auto/auto_factory.py", line 564, in from_pretrained
return model_class.from_pretrained(
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/modeling_utils.py", line 308, in _wrapper
return func(*args, **kwargs)
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/modeling_utils.py", line 4613, in from_pretrained
) = cls._load_pretrained_model(
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/modeling_utils.py", line 4923, in _load_pretrained_model
model._initialize_missing_keys(checkpoint_keys, ignore_mismatched_sizes, is_quantized)
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/modeling_utils.py", line 5516, in _initialize_missing_keys
self.initialize_weights()
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/modeling_utils.py", line 2554, in initialize_weights
self.smart_apply(self._initialize_weights)
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/transformers/modeling_utils.py", line 2545, in smart_apply
module.smart_apply(module._initialize_weights)
File "/home/robot/.conda/envs/mimic2/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1956, in getattr
raise AttributeError(
AttributeError: 'DaViT' object has no attribute '_initialize_weights'
How can i do to solve it,thanks for replying.