Here we show how to use FlexGen for the data wrangling tasks. The implementation follows the fm_data_tasks repo from HazyResearch.
- Run our install.sh scripts to install additional python library and download the desired datasets.
-
To check the outcome and verify the result of a data imputation task (e.g., Restaurant), run:
bash test_single_query_case.sh
-
To test FlexGen Throughput of a data imputation task (e.g., Restaurant), run:
bash test_batch_query.case.sh