BackTo2014 commited on
Commit
6dd4c1e
·
verified ·
1 Parent(s): 65f08f7

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "UNet2DModel",
3
  "T": 1000,
4
  "ch": 128,
5
  "ch_mult": [1, 2, 3, 4],
 
1
  {
2
+ "_class_name": "UNet",
3
  "T": 1000,
4
  "ch": 128,
5
  "ch_mult": [1, 2, 3, 4],