From 12eef9cca59da7dca4251ceed776e6eb26c2a7b9 Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Fri, 15 Dec 2023 18:49:58 +0100 Subject: [PATCH] remove default hf_repo from config --- configs/finetune-textual-inversion.toml | 2 +- src/refiners/training_utils/huggingface_datasets.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/finetune-textual-inversion.toml b/configs/finetune-textual-inversion.toml index 90597a8..b45853b 100644 --- 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 index 6433323..3e73ad8 100644 --- 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