search

LengSicong / Tell2Design

[ACL2023 Area Chair Award] Official repo for the paper "Tell2Design: A Dataset for Language-Guided Floor Plan Generation".
48 stars 1 forks source link

conflict with datasets module?? module 'datasets' has no attribute 'Dataset' in transfomers.Trainer #15

Open yusuke0324 opened 1 week ago

yusuke0324 commented 1 week ago

Hi,

I tried to train T5 model with Data/Separated\ Data/1st_finetune_data/T5/floorplan_train_wo_human.json (so I put copied to data/floorplan/floorplan_train.json) then ran $python run.py floorplan under T5 folder.

I got the error bellow.

line 638, in get_train_dataloader
    if is_datasets_available() and isinstance(train_dataset, datasets.Dataset):

transformers version is 4.14.1 (as mentioned in README)

I'm assuming transformers looked at T5/datasets.py instead of datasets module from Huggin face.

Is there anyone know how to work around this?

Another possible issue is module versions. Since the requirements.txt in this repo raised a lot of errors when they are pip installed, I rewrote it to minimize the list.