Upload VibeVoiceForConditionalGeneration
Browse files- config.json +5 -0
config.json
CHANGED
|
@@ -48,6 +48,11 @@
|
|
| 48 |
"mlp_bias": false,
|
| 49 |
"model_type": "vibevoice",
|
| 50 |
"noise_scheduler_class": "DPMSolverMultistepScheduler",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
"num_diffusion_steps": 10,
|
| 52 |
"num_head_layers": 4,
|
| 53 |
"pad_token_id": 151643,
|
|
|
|
| 48 |
"mlp_bias": false,
|
| 49 |
"model_type": "vibevoice",
|
| 50 |
"noise_scheduler_class": "DPMSolverMultistepScheduler",
|
| 51 |
+
"noise_scheduler_config": {
|
| 52 |
+
"beta_schedule": "squaredcos_cap_v2",
|
| 53 |
+
"num_train_timesteps": 1000,
|
| 54 |
+
"prediction_type": "v_prediction"
|
| 55 |
+
},
|
| 56 |
"num_diffusion_steps": 10,
|
| 57 |
"num_head_layers": 4,
|
| 58 |
"pad_token_id": 151643,
|