remove default hf_repo from config

finegrain-ai · Dec 20, 2023 · 43ca0f2 · 43ca0f2
1 parent dc123ec
commit 43ca0f2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/finetune-textual-inversion.toml b/configs/finetune-textual-inversion.toml
@@ -44,7 +44,7 @@ dropout_probability = 0
 use_gyro_dropout = false
 
 [dataset]
-hf_repo = "acme/cat-toy"
+hf_repo = "acme/images"
 revision = "main"
 horizontal_flip = true
 random_crop = true

diff --git a/src/refiners/training_utils/huggingface_datasets.py b/src/refiners/training_utils/huggingface_datasets.py
@@ -26,7 +26,7 @@ def load_hf_dataset(
 
 
 class HuggingfaceDatasetConfig(BaseModel):
-    hf_repo: str = "finegrain/unsplash-dummy"
+    hf_repo: str
     revision: str = "main"
     split: str = "train"
     horizontal_flip: bool = False