athenlab/mai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reduce diffusiondb sample size

Browse Source
This commit is contained in:
Kenneth
2024-05-17 19:40:46 +01:00
parent 47b0b2e7a8
commit eaa6cf38ca
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -33,7 +33,7 @@ def load_data():
diffusion_db_dataset = load_dataset( diffusion_db_dataset = load_dataset(
"poloclub/diffusiondb", "poloclub/diffusiondb",
"2m_random_50k", "2m_random_10k",
trust_remote_code=True, trust_remote_code=True,
split="train", split="train",
) )