Commit
•
495da4a
1
Parent(s):
47e682c
Update generation config (#6)
Browse files- update generation config (30e8e35e7874eccf9f53bbf49ae81e83f2824285)
- update readme (3b8e56eda940b295de957d4ec287e5f5be1190c9)
- 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 |
}
|