create-caption

Paused

nroggendorff commited on Nov 16, 2025

Commit

3f33161

verified ·

1 Parent(s): 9936b41

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -123,8 +123,8 @@ def process_shard_worker(
 def main():
-    input_dataset = "none-yet/anime-captions"
-    output_dataset = "nroggendorff/anime-captions"
     model_name = "datalab-to/chandra"
     batch_size = 32

 def main():
+    input_dataset = "none-yet/wikiart"
+    output_dataset = "nroggendorff/wikiart"
     model_name = "datalab-to/chandra"
     batch_size = 32