# get data from hugging face
ds = load_dataset("HuggingFaceH4/ultrachat_200k",split="train_gen")