allow `dataset_name` to be set outside and set the default value to `gist_80M`