AttributeError: module 'dlimp' has no attribute 'DLataset' #11
Description
Install repo
!git clone https://github.com/rail-berkeley/crossformer.git
%cd crossformer
!pip3 install -e .
!pip3 install -r requirements.txt
!pip3 install --upgrade "jax[cuda11_pip]==0.4.20" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
!pip install -U jax
pip install tensorflow tensorflow_datasets "numpy<2"
Installing Dlimp
!git clone https://github.com/kvablack/dlimp
Using pretrained model
from crossformer.model.crossformer_model import CrossFormerModel
model = CrossFormerModel.load_pretrained("hf://rail-berkeley/crossformer")
Error Log
AttributeError Traceback (most recent call last)
/tmp/ipython-input-1957569514.py in <cell line: 0>()
----> 1 from crossformer.model.crossformer_model import CrossFormerModel
2
3 model = CrossFormerModel.load_pretrained("hf://rail-berkeley/crossformer")
1 frames
/content/crossformer/crossformer/model/crossformer_model.py in
15 import tensorflow as tf
16
---> 17 from crossformer.data.utils.data_utils import NormalizationType
18 from crossformer.data.utils.text_processing import TextProcessor
19 from crossformer.model.components.action_heads import ActionHead
/content/crossformer/crossformer/data/utils/data_utils.py in
84
85 def get_dataset_statistics(
---> 86 dataset: dl.DLataset,
87 proprio_keys: list,
88 hash_dependencies: Tuple[str, ...],
AttributeError: module 'dlimp' has no attribute 'DLataset'