mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
remove default hf_repo from config
This commit is contained in:
parent
6a1fac876b
commit
12eef9cca5
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue