Natooz commited on
Commit
ce47774
1 Parent(s): 5e790d5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -42,9 +42,7 @@ The model is designed for autoregressive music generation. It generates the cont
42
  ## How to Get Started with the Model
43
 
44
  Use the code below to get started with the model.
45
- You will need the `miditok`, `transformers` and `torch` packages to make it run, that can be installed with pip.
46
-
47
- You will also need to manually download the `tokenizer.conf` file from the [repo files](https://huggingface.co/Natooz/Maestro-TSD-bpe20k/tree/main).
48
 
49
  ```Python
50
  import torch
@@ -54,7 +52,7 @@ from miditoolkit import MidiFile
54
 
55
  torch.set_default_device("cuda")
56
  model = AutoModelForCausalLM.from_pretrained("Natooz/Maestro-TSD-bpe20k", trust_remote_code=True, torch_dtype="auto")
57
- tokenizer = TSD(params="tokenizer.conf")
58
  input_midi = MidiFile("path/to/file.mid")
59
  input_tokens = tokenizer(input_midi)
60
 
 
42
  ## How to Get Started with the Model
43
 
44
  Use the code below to get started with the model.
45
+ You will need the `miditok` (>=v2.1.7), `transformers` and `torch` packages to make it run, that can be installed with pip.
 
 
46
 
47
  ```Python
48
  import torch
 
52
 
53
  torch.set_default_device("cuda")
54
  model = AutoModelForCausalLM.from_pretrained("Natooz/Maestro-TSD-bpe20k", trust_remote_code=True, torch_dtype="auto")
55
+ tokenizer = TSD.from_pretrained("Natooz/Maestro-TSD-bpe20k")
56
  input_midi = MidiFile("path/to/file.mid")
57
  input_tokens = tokenizer(input_midi)
58