search

akaashdash / xlstm

MIT License
24 stars 0 forks source link

I am not being able to import xlstm #3

Closed sm-ak-r33 closed 1 month ago

sm-ak-r33 commented 1 month ago

I did '%pip install xlstm' yet 'from xlstm import xLSTM' is giving me

ModuleNotFoundError Traceback (most recent call last) Cell In[15], line 7 5 import torch.optim as optim 6 from torch.utils.data import DataLoader, Dataset ----> 7 from xlstm import xLSTM 9 # Define the dataset class 10 class TimeSeriesDataset(Dataset):

ModuleNotFoundError: No module named 'xlstm'

How to fix this?

akaashdash commented 1 month ago

The package is torchxlstm. https://pypi.org/project/torchxlstm/