Upload VibeVoiceForConditionalGeneration
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -47,9 +47,9 @@
|
|
| 47 |
"intermediate_size": 4608,
|
| 48 |
"mlp_bias": false,
|
| 49 |
"model_type": "vibevoice",
|
|
|
|
| 50 |
"noise_scheduler_config": {
|
| 51 |
"beta_schedule": "squaredcos_cap_v2",
|
| 52 |
-
"num_train_timesteps": 1000,
|
| 53 |
"prediction_type": "v_prediction"
|
| 54 |
},
|
| 55 |
"num_diffusion_steps": 10,
|
|
|
|
| 47 |
"intermediate_size": 4608,
|
| 48 |
"mlp_bias": false,
|
| 49 |
"model_type": "vibevoice",
|
| 50 |
+
"noise_scheduler_class": "DPMSolverMultistepScheduler",
|
| 51 |
"noise_scheduler_config": {
|
| 52 |
"beta_schedule": "squaredcos_cap_v2",
|
|
|
|
| 53 |
"prediction_type": "v_prediction"
|
| 54 |
},
|
| 55 |
"num_diffusion_steps": 10,
|