Spaces:
Runtime error
Runtime error
add datasets>=2.9.0 requirement
Browse files- requirements.txt +1 -1
- scripts/train_unet.py +3 -2
- setup.cfg +1 -1
requirements.txt
CHANGED
@@ -3,7 +3,7 @@ numpy
|
|
3 |
Pillow
|
4 |
diffusers>=0.12.0
|
5 |
librosa
|
6 |
-
datasets
|
7 |
gradio
|
8 |
streamlit
|
9 |
tensorboard
|
|
|
3 |
Pillow
|
4 |
diffusers>=0.12.0
|
5 |
librosa
|
6 |
+
datasets>=2.9.0
|
7 |
gradio
|
8 |
streamlit
|
9 |
tensorboard
|
scripts/train_unet.py
CHANGED
@@ -52,8 +52,9 @@ def main(args):
|
|
52 |
|
53 |
if args.dataset_name is not None:
|
54 |
if os.path.exists(args.dataset_name):
|
55 |
-
dataset = load_from_disk(
|
56 |
-
|
|
|
57 |
else:
|
58 |
dataset = load_dataset(
|
59 |
args.dataset_name,
|
|
|
52 |
|
53 |
if args.dataset_name is not None:
|
54 |
if os.path.exists(args.dataset_name):
|
55 |
+
dataset = load_from_disk(
|
56 |
+
args.dataset_name,
|
57 |
+
storage_options=args.dataset_config_name)["train"]
|
58 |
else:
|
59 |
dataset = load_dataset(
|
60 |
args.dataset_name,
|
setup.cfg
CHANGED
@@ -17,4 +17,4 @@ install_requires =
|
|
17 |
Pillow
|
18 |
diffusers>=0.12.0
|
19 |
librosa
|
20 |
-
datasets
|
|
|
17 |
Pillow
|
18 |
diffusers>=0.12.0
|
19 |
librosa
|
20 |
+
datasets>=2.9.0
|