Hugging Face's logo

Spaces:
teticio
/
audio-diffusion
Runtime error

App Files Community
1
teticio commited on
Commit
507b852
•
1 Parent(s): afdefbb

add datasets>=2.9.0 requirement

Browse files
Files changed (3) hide show
  1. requirements.txt +1 -1
  2. scripts/train_unet.py +3 -2
  3. 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(args.dataset_name,
56
- storage_options = args.dataset_config_name)["train"]
 
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