Skip to content

Commit

Permalink
Merge pull request #257 from facebookresearch/readme/cache_dir
Browse files Browse the repository at this point in the history
Update README
  • Loading branch information
JadeCopet authored Aug 29, 2023
2 parents b50c9a4 + 9c405e0 commit 3dea2b9
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,9 @@ Yes! We provide the training code for [EnCodec](./docs/ENCODEC.md), [MusicGen](.

#### Where are the models stored?

Hugging Face stored the model in a specific location, which can be overriden by setting the `AUDIOCRAFT_CACHE_DIR` environment variable.
Hugging Face stored the model in a specific location, which can be overriden by setting the `AUDIOCRAFT_CACHE_DIR` environment variable for the AudioCraft models.
In order to change the cache location of the other Hugging Face models, please check out the [Hugging Face Transformers documentation for the cache setup](https://huggingface.co/docs/transformers/installation#cache-setup).
Finally, if you use a model that relies on Demucs (e.g. `musicgen-melody`) and want to change the download location for Demucs, refer to the [Torch Hub documentation](https://pytorch.org/docs/stable/hub.html#where-are-my-downloaded-models-saved).


## License
Expand Down
9 changes: 9 additions & 0 deletions config/solver/audiogen/debug.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,15 @@ defaults:

autocast: false
compression_model_checkpoint: null
transformer_lm:
n_q: 4
card: 400

conditioners:
description:
model: t5
t5:
name: t5-small

codebooks_pattern:
modeling: parallel
Expand Down
6 changes: 6 additions & 0 deletions config/solver/musicgen/debug.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,12 @@ transformer_lm:
n_q: 4
card: 400

conditioners:
description:
model: t5
t5:
name: t5-small

codebooks_pattern:
modeling: parallel

Expand Down

0 comments on commit 3dea2b9

Please sign in to comment.