search

worldbank / REaLTabFormer

A suite of auto-regressive and Seq2Seq (sequence-to-sequence) transformer models for tabular and relational synthetic data generation.
https://worldbank.github.io/REaLTabFormer/
MIT License
203 stars 23 forks source link

Unable to complete training in colab #24

Closed vinay-k12 closed 1 year ago

vinay-k12 commented 1 year ago

Trying to run the model on california pricing tabular data. But continuously facing this issue even if I install the libraries.

image

Am I missing anything?

image
vinay-k12 commented 1 year ago

Found the issue. I need to restart colab post running !pip install accelerate -U

avsolatorio commented 1 year ago

@vinay-k12 thanks for letting us know about this! A new release v0.1.3 was published to include accelerate as a dependency. The colab notebook should again work now. 😀