From eaa6cf38ca4f3813e04fc0f1b1c0f50c30635af1 Mon Sep 17 00:00:00 2001
From: Kenneth <kennethnym@outlook.com>
Date: Fri, 17 May 2024 19:40:46 +0100
Subject: [PATCH] reduce diffusiondb sample size

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 72a46a2..506501d 100644
--- a/train.py
+++ b/train.py
@@ -33,7 +33,7 @@ def load_data():
 
     diffusion_db_dataset = load_dataset(
         "poloclub/diffusiondb",
-        "2m_random_50k",
+        "2m_random_10k",
         trust_remote_code=True,
         split="train",
     )