Skip to content

Scripts to generate TFDS format of Redpajama dataset.

Notifications You must be signed in to change notification settings

Orion-Zheng/redpajama_tfds

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Please replace the environmental variable $SCRATCH in all scripts with the actual path to redpajama_tfds on your machine
Dependencies: tfds-nightly, zstandard

1. Download raw data files

Run download_scripts\subsets_wget.sh, and then all raw data files will appear under the redpajama_raw directory.

2. Verify the Checksum and decompress common_crawl subset

  1. Run redpajama_raw\check_checksum.sh, making sure all downloaded data files are complete.
  2. Run redpajama_raw\decompress_cc.sh, decompressing common_crawl subset from .zst to .jsonl and delete original .zst files.

3. Generate TFDS of all redpajama subsets

Run run_build\generate_SUBSET_NAME.sh to generate TFDS of all subsets to the redpajama_data directory.
Among these scripts:

--manual_dir: The source directory for storing raw data.
--data_dir: The target directory for storing the generated TFDS.

4. Upload all TFDS to Google Cloud

Install gsutil and sign in your Google Account, run redpajama_data/upload.sh to upload TFDS to your storage bucket.
ref

About

Scripts to generate TFDS format of Redpajama dataset.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published