Commit
•
941ba65
1
Parent(s):
5bf13ed
Update generation config (#5)
Browse files- update generation config (cc3b291ccc9166a9a6694be052741112693ff910)
- update readme (8a08efe1c36001d8ca140cb2a723a69f2439e836)
- README.md +1 -1
- generation_config.json +3 -0
README.md
CHANGED
@@ -66,7 +66,7 @@ inputs = processor(
|
|
66 |
return_tensors="pt",
|
67 |
)
|
68 |
|
69 |
-
audio_values = model.generate(**inputs,
|
70 |
```
|
71 |
|
72 |
3. Listen to the audio samples either in an ipynb notebook:
|
|
|
66 |
return_tensors="pt",
|
67 |
)
|
68 |
|
69 |
+
audio_values = model.generate(**inputs, max_new_tokens=256)
|
70 |
```
|
71 |
|
72 |
3. Listen to the audio samples either in an ipynb notebook:
|
generation_config.json
CHANGED
@@ -2,6 +2,9 @@
|
|
2 |
"_from_model_config": true,
|
3 |
"bos_token_id": 2048,
|
4 |
"decoder_start_token_id": 2048,
|
|
|
|
|
|
|
5 |
"pad_token_id": 2048,
|
6 |
"transformers_version": "4.31.0.dev0"
|
7 |
}
|
|
|
2 |
"_from_model_config": true,
|
3 |
"bos_token_id": 2048,
|
4 |
"decoder_start_token_id": 2048,
|
5 |
+
"do_sample": true,
|
6 |
+
"guidance_scale": 3.0,
|
7 |
+
"max_length": 1500,
|
8 |
"pad_token_id": 2048,
|
9 |
"transformers_version": "4.31.0.dev0"
|
10 |
}
|