Hugging Face's logo

Spaces:
jbilcke-hf
/
ai-tube-model-musicgen-3
Running on A10G

App Files Community
ai-tube-model-musicgen-3
File size: 812 Bytes 
5325fcc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
 
# @package __global__

defaults:
  - compression/default
  - /model: encodec/encodec_base_causal
  - override /dset: audio/example
  - _self_

channels: 1
sample_rate: 16000

# debug config uses just L1
losses:
  adv: 0.
  feat: 0.
  l1: 1.
  mel: 0.
  msspec: 0.
# no balancer
balancer:
  balance_grads: false
  ema_decay: 1.
  total_norm: 1.
  per_batch_item: false
# no adversaries
adversarial:
  adversaries: []
  adv_loss: hinge
  feat_loss: l1

# faster model for local dev
seanet:
  dimension: 16
  n_filters: 4

# very small dataset
dataset:
  batch_size: 8
  num_workers: 10
  num_samples: 100
  segment_duration: 1
  evaluate:
    batch_size: 32
  generate:
    batch_size: 1
    num_samples: 5
    segment_duration: 10

# limited training
evaluate:
  every: 5
generate:
  every: 5
optim:
  epochs: 50