config: map AutoModelForTextToWaveform so pipeline() resolves
#3
by nullHawk - opened
- config.json +2 -1
config.json
CHANGED
|
@@ -9,7 +9,8 @@
|
|
| 9 |
"audio_start_token_id": 277393,
|
| 10 |
"auto_map": {
|
| 11 |
"AutoConfig": "configuration_rumik_oss.RumikOSSConfig",
|
| 12 |
-
"AutoModelForCausalLM": "modeling_rumik_oss.RumikOSSForCausalLM"
|
|
|
|
| 13 |
},
|
| 14 |
"bos_token_id": 2,
|
| 15 |
"codebook_size": 2048,
|
|
|
|
| 9 |
"audio_start_token_id": 277393,
|
| 10 |
"auto_map": {
|
| 11 |
"AutoConfig": "configuration_rumik_oss.RumikOSSConfig",
|
| 12 |
+
"AutoModelForCausalLM": "modeling_rumik_oss.RumikOSSForCausalLM",
|
| 13 |
+
"AutoModelForTextToWaveform": "modeling_rumik_oss.RumikOSSForCausalLM"
|
| 14 |
},
|
| 15 |
"bos_token_id": 2,
|
| 16 |
"codebook_size": 2048,
|