Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-10 07:21:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove default hf_repo from config

Browse source
This commit is contained in:
limiteinductive 2023-12-15 18:49:58 +01:00 committed by Benjamin Trom
parent 6a1fac876b
commit 12eef9cca5
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
configs/finetune-textual-inversion.toml
View file

@ -44,7 +44,7 @@ dropout_probability = 0
use_gyro_dropout = false use_gyro_dropout = false
[dataset] [dataset]
hf_repo = "acme/cat-toy" hf_repo = "acme/images"
revision = "main" revision = "main"
horizontal_flip = true horizontal_flip = true
random_crop = true random_crop = true

2
src/refiners/training_utils/huggingface_datasets.py
View file

@ -26,7 +26,7 @@ def load_hf_dataset(
class HuggingfaceDatasetConfig(BaseModel): class HuggingfaceDatasetConfig(BaseModel):
hf_repo: str = "finegrain/unsplash-dummy" hf_repo: str
revision: str = "main" revision: str = "main"
split: str = "train" split: str = "train"
horizontal_flip: bool = False horizontal_flip: bool = False