anton-l HF staff commited on
Commit
625a079
1 Parent(s): bcfacd2

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
  }