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