Update text_noise_scheduler/scheduler_config.json
Browse files
text_noise_scheduler/scheduler_config.json
CHANGED
@@ -2,5 +2,6 @@
|
|
2 |
"_class_name": "DDPMScheduler",
|
3 |
"_diffusers_version": "0.0.3",
|
4 |
"beta_schedule": "squaredcos_cap_v2",
|
5 |
-
"timesteps": 1000
|
|
|
6 |
}
|
|
|
2 |
"_class_name": "DDPMScheduler",
|
3 |
"_diffusers_version": "0.0.3",
|
4 |
"beta_schedule": "squaredcos_cap_v2",
|
5 |
+
"timesteps": 1000,
|
6 |
+
"tensor_format": "pt"
|
7 |
}
|