Hugging Face's logo

archinetai
/
dmae1d-ATC64-v2

Feature Extraction
Transformers
PyTorch
archinetai/dmae1d-ATC64-v2
custom_code
Model card Files Community
flavioschneider commited on
Commit
3ffeea6
•
1 Parent(s): 7cd616c

Upload DMAE1d

Browse files
Files changed (2) hide show
  1. config.json +6 -1
  2. dmae.py +1 -2
config.json CHANGED
@@ -1,7 +1,12 @@
1
  {
 
 
 
2
  "auto_map": {
3
- "AutoConfig": "dmae_config.DMAE1dConfig"
 
4
  },
5
  "model_type": "archinetai/dmae1d-ATC64-v2",
 
6
  "transformers_version": "4.24.0"
7
  }
 
1
  {
2
+ "architectures": [
3
+ "DMAE1d"
4
+ ],
5
  "auto_map": {
6
+ "AutoConfig": "dmae_config.DMAE1dConfig",
7
+ "AutoModel": "dmae.DMAE1d"
8
  },
9
  "model_type": "archinetai/dmae1d-ATC64-v2",
10
+ "torch_dtype": "float32",
11
  "transformers_version": "4.24.0"
12
  }
dmae.py CHANGED
@@ -2,8 +2,7 @@ import torch
2
  from transformers import PreTrainedModel
3
  from .dmae_config import DMAE1dConfig
4
  from audio_encoders_pytorch import ME1d, TanhBottleneck # pip install audio_encoders_pytorch==0.0.20
5
- from audio_diffusion_pytorch.unets import UNetV0, LTPlugin # pip install -U git+https://github.com/archinetai/audio-diffusion-pytorch.git@nightly # v0.0.2
6
- from audio_diffusion_pytorch.models import DiffusionAE
7
 
8
  class DMAE1d(PreTrainedModel):
9
 
 
2
  from transformers import PreTrainedModel
3
  from .dmae_config import DMAE1dConfig
4
  from audio_encoders_pytorch import ME1d, TanhBottleneck # pip install audio_encoders_pytorch==0.0.20
5
+ from audio_diffusion_pytorch import UNetV0, LTPlugin, DiffusionAE
 
6
 
7
  class DMAE1d(PreTrainedModel):
8