Commit
•
f65fe41
1
Parent(s):
e5c69f2
Update generation config (#3)
Browse files- update generation config (b93459a35f80c2985009cdf9789e8458a4ff3698)
- update readme (aef8e3f57b55578fcea41d17a1d44321dd3dcb1b)
- 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 |
}
|